May 6, 2026 · cond-mat.mtrl-sciJ/K move · Enter open · S save
Rafael Espinosa Castañeda, Ashley Dale, Hongchen Wang, Yonatan Kurniawan+5
Department of Materials Science and Engineering, University of Toronto, Canada. · 4Vector Institute for Artificial Intelligence, Toronto, Canada. · Department of Computer Science, Princeton University · Department of Materials Science and Applied Physics,May King Abdullah University of Science and Technology, Saudi Arabia. · 5Acceleration Consortium, Toronto, Canada
Materials science data collection can be expensive, making the reuse and long-term utility of datasets critical important for future discovery campaigns. In practice, researchers prioritize a subset of properties due to research interests. However, ignoring a subset of outcomes in data collection campaigns potentially generate datasets poorly suited for future learning tasks. Here, we present a framework for dataset construction that maximizes informativeness for target properties of interest while preserving performance on untargeted ones. Our approach uses diversity-aware selection to ensure broad coverage of the materials space. In noisy experimental dataset construction, we find that without our diversity-aware framework, prediction performance on untargeted properties can degrade by up to 40% relative to random sampling, whereas applying our framework yields improvements of up to 10% . For targeted properties, performance can degrade with respect to random sampling by up to 12.5% without diversity, while our framework achieves gains of up to 25%. Incorporating diversity into dataset construction not only preserves informativeness for the targeted properties, but also improves materials coverage for potential future objectives. As a result, the constructed datasets remain broadly informative across considered and unconsidered outcomes, ensuring unbiased quality entries and mitigating cold-start limitations in subsequent modeling and discovery campaigns.