Multi-Fidelity Active Learning with GFlowNets
Authors that are also TMLR Expert Reviewers: ~Yoshua_Bengio1
Abstract: In the last decades, the capacity to generate large amounts of data in science and engineering applications has been growing steadily. Meanwhile, machine learning has progressed to become a suitable tool to process and utilise the available data. Nonetheless, many relevant scientific and engineering problems present challenges where current machine learning methods cannot yet efficiently leverage the available data and resources. For example, in scientific discovery, we are often faced with the problem of exploring very large, structured and high-dimensional spaces. Moreover, the high fidelity, black-box objective function is often very expensive to evaluate. Progress in machine learning methods that can efficiently tackle such challenges would help accelerate currently crucial areas such as drug and materials discovery. In this paper, we propose a multi-fidelity active learning algorithm with GFlowNets as a sampler, to efficiently discover diverse, high-scoring candidates where multiple approximations of the black-box function are available at lower fidelity and cost. Our evaluation on molecular discovery tasks shows that multi-fidelity active learning with GFlowNets can discover high-scoring candidates at a fraction of the budget of its single-fidelity counterpart while maintaining diversity, unlike RL-based alternatives. These results open new avenues for multi-fidelity active learning to accelerate scientific discovery and engineering design.
Certifications: Expert Certification
Submission Length: Regular submission (no more than 12 pages of main content)
Changes Since Last Submission: - Proof reading
- Update and harmonise tables
- Make notation more consistent
- Harmonise bibliography
- Better description of oracle confidences
Video: https://www.dailymotion.com/video/k1k8KKYS67DgFCB516w
Code: https://github.com/nikita-0209/mf-al-gfn/
Supplementary Material: zip
Assigned Action Editor: ~Andreas_Kirsch1
Submission Number: 2400
Loading