Tackling the most pressing problems for humanity, such as the climate crisis and the threat of global pandemics, requires accelerating the pace of scientific discovery. While science has traditionally relied on trial and error and even serendipity to a large extent, the last few decades have seen a surge of data-driven scientific discoveries. However, in order to truly leverage large-scale data sets and high-throughput experimental setups, machine learning methods will need to be further improved and better integrated in the scientific discovery pipeline. A key challenge for current machine learning methods in this context is the efficient exploration of very large search spaces, which requires techniques for estimating reducible (epistemic) uncertainty and generating sets of diverse and informative experiments to perform. This motivated a new probabilistic machine learning framework called GFlowNets, which can be applied in the modeling, hypotheses generation and experimental design stages of the experimental science loop. GFlowNets learn to sample from a distribution given indirectly by a reward function corresponding to an unnormalized probability, which enables sampling diverse, high-reward candidates. GFlowNets can also be used to form efficient and amortized Bayesian posterior estimators for causal models conditioned on the already acquired experimental data. Having such posterior models can then provide estimators of epistemic uncertainty and information gain that can drive an experimental design policy. Altogether, here we will argue that GFlowNets can become a valuable tool for AI-driven scientific discovery, especially in scenarios of very large candidate spaces where we have access to cheap but inaccurate measurements or to expensive but accurate measurements. This is a common setting in the context of drug and material discovery, which we use as examples throughout the paper.
翻译:应对人类最紧迫的问题,如气候危机和全球大流行病的威胁,需要加快科学发现的步伐。尽管科学传统上在很大程度上依赖试错甚至偶然发现,但过去几十年见证了数据驱动科学发现的激增。然而,为了真正利用大规模数据集和高通量实验设置,机器学习方法需要进一步改进并更好地融入科学发现流程。在此背景下,当前机器学习方法的一个关键挑战是如何高效探索极大的搜索空间,这需要技术来估计可缩减的(认知性)不确定性,并生成多样化且信息丰富的实验方案。这促使了一种名为GFlowNets的新型概率机器学习框架的诞生,该框架可应用于实验科学循环中的建模、假设生成和实验设计阶段。GFlowNets学习从由未归一化概率对应的奖励函数间接给定的分布中进行采样,从而能够采样多样化且高奖励的候选方案。GFlowNets还可用于形成高效且摊销化的贝叶斯后验估计器,以针对已获取实验数据条件下的因果模型。拥有这样的后验模型后,便可提供认知性不确定性和信息增益的估计值,进而驱动实验设计策略。总之,本文认为GFlowNets可成为人工智能驱动科学发现的宝贵工具,尤其是在候选空间极大且我们可获取廉价但不精确的测量或昂贵但精确的测量时。这种情形在药物和材料发现中很常见,本文便以此为例进行阐述。