{"id":"bayesian-active-learning","title":"Bayesian Active Learning for Potential Selection","subtitle":"Gaussian Process surrogates across 23 potentials × 12,000 materials.","category":"methods","tags":["bayesian","active-learning"],"source":"articles/docs/bayesian_active_learning_report.md","lang":"en","words":958,"readMinutes":4,"toc":[{"depth":2,"text":"Kimi Deep Research Report — Comprehensive Literature Review & Practical Guide","id":"kimi-deep-research-report-comprehensive-literature-review-practical-guide"},{"depth":2,"text":"Report Structure (10 Major Sections, 22+ Tables)","id":"report-structure-10-major-sections-22-tables"},{"depth":3,"text":"1. Problem Formulation and Theoretical Foundations","id":"1-problem-formulation-and-theoretical-foundations"},{"depth":3,"text":"2. Bayesian Model Averaging (BMA) for Multi-Potential Ensembles","id":"2-bayesian-model-averaging-bma-for-multi-potential-ensembles"},{"depth":3,"text":"3. Gaussian Process Surrogates for Error Prediction","id":"3-gaussian-process-surrogates-for-error-prediction"},{"depth":3,"text":"4. Active Learning and Query-by-Committee for DFT Prioritization","id":"4-active-learning-and-query-by-committee-for-dft-prioritization"},{"depth":3,"text":"5. Multi-Fidelity Bayesian Optimization","id":"5-multi-fidelity-bayesian-optimization"},{"depth":3,"text":"6. Domain-Specific Considerations for Crystalline Materials","id":"6-domain-specific-considerations-for-crystalline-materials"},{"depth":3,"text":"7. Gaussian Process Architecture Design for GLIM","id":"7-gaussian-process-architecture-design-for-glim"},{"depth":3,"text":"8. Literature Synthesis: Key References and Methods","id":"8-literature-synthesis-key-references-and-methods"},{"depth":3,"text":"9. Comparative Analysis of Acquisition Strategies (Tables 18-21)","id":"9-comparative-analysis-of-acquisition-strategies-tables-18-21"},{"depth":3,"text":"10. Implementation Roadmap","id":"10-implementation-roadmap"},{"depth":2,"text":"Key Findings for GLIM","id":"key-findings-for-glim"}],"html":"<h1 id=\"bayesian-active-learning-for-interatomic-potential-selection\">Bayesian Active Learning for Interatomic Potential Selection</h1><h2 id=\"kimi-deep-research-report-comprehensive-literature-review-amp-practical-guide\">Kimi Deep Research Report — Comprehensive Literature Review &amp; Practical Guide</h2><p><strong>Tags:</strong> Bayesian Optimization | Active Learning | Materials Science | GLIM</p>\n<hr>\n<h2 id=\"report-structure-10-major-sections-22-tables\">Report Structure (10 Major Sections, 22+ Tables)</h2><h3 id=\"1-problem-formulation-and-theoretical-foundations\">1. Problem Formulation and Theoretical Foundations</h3><h4 id=\"1-1-the-glim-benchmark-challenge\">1.1 The GLIM Benchmark Challenge</h4><h5 id=\"1-1-1-dataset-characteristics-23-potentials-x-12-000-materials\">1.1.1 Dataset Characteristics: 23 Potentials x 12,000 Materials</h5><p>The GLIM benchmark represents a paradigm shift in computational materials science, providing systematic evaluation of 23 distinct interatomic potentials across approximately 12,000 materials from the JARVIS-FF database. This massive-scale comparison creates a performance matrix of 276,000 potential-material combinations for each target property. The 23 potentials encompass a diverse methodological spectrum: classical empirical potentials (EAM, MEAM, ReaxFF), machine learning interatomic potentials (SNAP, GAP, neural network potentials), and recent universal pre-trained models (M3GNet, CHGNet, MACE-MP0).</p>\n<p><strong>Table 1</strong>: Taxonomy of interatomic potentials in the GLIM benchmark, illustrating the cost-accuracy spectrum that motivates multi-fidelity optimization.</p>\n<h5 id=\"1-1-2-target-properties-bulk-modulus-k-shear-moduli-c-39-c44\">1.1.2 Target Properties: Bulk Modulus (K), Shear Moduli (C&#39;, C44)</h5><p><strong>Table 2</strong>: Target elastic constants in the GLIM benchmark, showing their complementary information content and varying sensitivity to potential quality.</p>\n<h5 id=\"1-1-3-dimensionality-reduction-pca-analysis-with-55-first-component-variance\">1.1.3 Dimensionality Reduction: PCA Analysis with 55% First-Component Variance</h5><p>The PCA decomposition operates on the centered prediction error matrix, where entries represent deviations from DFT reference values. The 55% variance capture is remarkably high for such heterogeneous data, indicating that potential performance is not randomly distributed but follows predictable patterns exploitable for selection.</p>\n<div class=\"table-wrap\"><table><thead><tr>\n<th>Principal Component</th>\n<th>Variance Explained</th>\n<th>Interpretation</th>\n<th>Implications for Selection</th>\n</tr>\n</thead><tbody><tr>\n<td data-label=\"Principal Component\">PC1</td>\n<td data-label=\"Variance Explained\">55%</td>\n<td data-label=\"Interpretation\">Global &quot;potential quality&quot; / &quot;material difficulty&quot; axis</td>\n<td data-label=\"Implications for Selection\">Coarse ranking; identifies broadly good/bad potentials</td>\n</tr>\n<tr>\n<td data-label=\"Principal Component\">PC2</td>\n<td data-label=\"Variance Explained\">~15%</td>\n<td data-label=\"Interpretation\">Ionic vs. covalent bonding discrimination</td>\n<td data-label=\"Implications for Selection\">Separates potential families by bonding type</td>\n</tr>\n<tr>\n<td data-label=\"Principal Component\">PC3</td>\n<td data-label=\"Variance Explained\">~10%</td>\n<td data-label=\"Interpretation\">Crystal structure effects (fcc vs. bcc vs. hcp)</td>\n<td data-label=\"Implications for Selection\">Structure-specific performance patterns</td>\n</tr>\n<tr>\n<td data-label=\"Principal Component\">PC4-5</td>\n<td data-label=\"Variance Explained\">~5% each</td>\n<td data-label=\"Interpretation\">Property-specific effects</td>\n<td data-label=\"Implications for Selection\">Fine-grained, property-dependent selection</td>\n</tr>\n</tbody></table></div><h5 id=\"1-1-4-core-decision-problem-material-specific-potential-trustworthiness\">1.1.4 Core Decision Problem: Material-Specific Potential Trustworthiness</h5><p><strong>Table 4</strong>: Taxonomy of uncertainty sources in potential selection, distinguishing reducible (epistemic) from irreducible (aleatoric) components.</p>\n<h4 id=\"1-2-statistical-framework-for-model-selection\">1.2 Statistical Framework for Model Selection</h4><ul>\n<li>1.2.1 From Single-Model Prediction to Ensemble Decision-Making</li>\n<li>1.2.2 Connection to Multi-Armed Bandit Theory</li>\n<li>1.2.3 Bayesian Information Criteria for Potential Ranking</li>\n</ul>\n<hr>\n<h3 id=\"2-bayesian-model-averaging-bma-for-multi-potential-ensembles\">2. Bayesian Model Averaging (BMA) for Multi-Potential Ensembles</h3><h4 id=\"2-1-classical-bma-theory-adapted-to-interatomic-potentials\">2.1 Classical BMA Theory Adapted to Interatomic Potentials</h4><ul>\n<li>2.1.1 Posterior Weight Computation via Marginal Likelihood</li>\n<li>2.1.2 Heteroscedastic Error Models Capturing Material-Dependent Accuracy</li>\n<li>2.1.3 Property-Dependent Weight Profiles: Why &quot;Best Overall&quot; != &quot;Best for K&quot;</li>\n</ul>\n<p>Specialist behavior is valuable for application-targeted selection — the same material may warrant different potentials for bulk modulus screening versus dislocation modeling.</p>\n<h4 id=\"2-2-implementation-strategies-for-interatomic-potential-ensembles\">2.2 Implementation Strategies for Interatomic Potential Ensembles</h4><h5 id=\"2-2-1-training-set-likelihood-evaluation-across-jarvis-ff-materials\">2.2.1 Training Set Likelihood Evaluation Across JARVIS-FF Materials</h5><p>Practical BMA implementation requires efficient likelihood computation at GLIM scale. The heteroscedastic error model:</p>\n<p>log p(y_i | m, M_k) ~ -(y_i - y_ki)^2 / (2 sigma_k^2(m_i)) - log sigma_k(m_i)</p>\n<p>The material-dependent variance sigma_k^2(m) is learned via GP regression on descriptors, enabling location-adaptive uncertainty.</p>\n<hr>\n<h3 id=\"3-gaussian-process-surrogates-for-error-prediction\">3. Gaussian Process Surrogates for Error Prediction</h3><ul>\n<li>3.1 Multi-Output GP Framework for the GLIM Error Matrix</li>\n<li>3.2 Kernel Design for Material-Potential Interactions (Tables 8-9)</li>\n<li>3.3 Scalable GP Approximations for 12,000-Material Problems (Table 10)</li>\n</ul>\n<hr>\n<h3 id=\"4-active-learning-and-query-by-committee-for-dft-prioritization\">4. Active Learning and Query-by-Committee for DFT Prioritization</h3><h4 id=\"4-1-ensemble-disagreement-as-uncertainty-measure\">4.1 Ensemble Disagreement as Uncertainty Measure</h4><h5 id=\"4-1-2-jensen-shannon-divergence-for-multi-potential-comparison\">4.1.2 Jensen-Shannon Divergence for Multi-Potential Comparison</h5><p>JSD(m) = (1/K) sum_k KL(p_k(.|m) || p_ens(.|m))</p>\n<p>Captures both mean and variance disagreement, identifying materials where potentials differ in confidence as well as point estimates.</p>\n<h5 id=\"4-1-3-committee-variance-as-proxy-for-epistemic-uncertainty\">4.1.3 Committee Variance as Proxy for Epistemic Uncertainty</h5><p>sigma^2_committee(m) = (1/(K-1)) sum_k (y_k(m) - y_bar(m))^2</p>\n<p>Decomposition: Total error^2 = Bias^2 + sigma^2_committee + sigma^2_aleatoric. The committee variance term is reducible epistemic uncertainty — the target of active learning.</p>\n<h4 id=\"4-2-uncertainty-driven-dft-prioritization\">4.2 Uncertainty-Driven DFT Prioritization</h4><h5 id=\"4-2-1-maximum-expected-prediction-error-mepe-criteria\">4.2.1 Maximum Expected Prediction Error (MEPE) Criteria</h5><p>alpha_MEPE(m) = mu^2_bias(m) + sigma^2_committee(m)</p>\n<ul>\n<li>4.2.2 Acquisition Function Design for Multi-Potential Selection (Table 12)</li>\n<li>4.2.3 Batch Selection Strategies for Parallel DFT Computation</li>\n</ul>\n<hr>\n<h3 id=\"5-multi-fidelity-bayesian-optimization\">5. Multi-Fidelity Bayesian Optimization</h3><ul>\n<li>5.1 Cost-Accuracy Hierarchy Among 23 Potentials (Table 13)</li>\n<li>5.2 Multi-Fidelity GP Models (MFGP) (Table 14)</li>\n<li>5.3 Information-Theoretic Acquisition Functions (alpha-UA)</li>\n</ul>\n<hr>\n<h3 id=\"6-domain-specific-considerations-for-crystalline-materials\">6. Domain-Specific Considerations for Crystalline Materials</h3><ul>\n<li>6.1 Crystal Symmetry-Aware Feature Engineering (Table 15)</li>\n<li>6.2 Composition-Structure-Property Relationships in Feature Space</li>\n<li>6.3 Transfer Learning Between Crystal Structure Families</li>\n</ul>\n<hr>\n<h3 id=\"7-gaussian-process-architecture-design-for-glim\">7. Gaussian Process Architecture Design for GLIM</h3><ul>\n<li>7.1.1 Matern-5/2 recommended: balance between smoothness and flexibility</li>\n<li>7.1.2 Heteroscedastic GPs for Input-Dependent Noise: Twin GP formulation</li>\n<li>7.1.3 Warped GPs for Non-Gaussian Error Distributions</li>\n</ul>\n<p><strong>Table 17</strong>: Kernel selection for GP surrogates in materials applications.</p>\n<hr>\n<h3 id=\"8-literature-synthesis-key-references-and-methods\">8. Literature Synthesis: Key References and Methods</h3><ul>\n<li>GP-MFBO, MEPE, UAPCA, alpha-UA acquisition function</li>\n<li>MC dropout for atomistic ML uncertainty</li>\n<li>QBC ensemble disagreement methods</li>\n<li>Bayesian model selection for elastic materials</li>\n<li>Ensemble variance as most promising acquisition criterion</li>\n</ul>\n<hr>\n<h3 id=\"9-comparative-analysis-of-acquisition-strategies-tables-18-21\">9. Comparative Analysis of Acquisition Strategies (Tables 18-21)</h3><hr>\n<h3 id=\"10-implementation-roadmap\">10. Implementation Roadmap</h3><h4 id=\"10-1-software-ecosystem\">10.1 Software Ecosystem</h4><div class=\"table-wrap\"><table><thead><tr>\n<th>Tool</th>\n<th>Role</th>\n<th>Key Features</th>\n</tr>\n</thead><tbody><tr>\n<td data-label=\"Tool\">BoTorch</td>\n<td data-label=\"Role\">Bayesian optimization</td>\n<td data-label=\"Key Features\">GPU-accelerated, modular acquisition functions</td>\n</tr>\n<tr>\n<td data-label=\"Tool\">GPyTorch</td>\n<td data-label=\"Role\">GP modeling</td>\n<td data-label=\"Key Features\">Scalable exact/approximate inference</td>\n</tr>\n<tr>\n<td data-label=\"Tool\">ASE</td>\n<td data-label=\"Role\">Atomistic simulation</td>\n<td data-label=\"Key Features\">Calculator interface for all potentials</td>\n</tr>\n<tr>\n<td data-label=\"Tool\">pymatgen</td>\n<td data-label=\"Role\">Structure analysis</td>\n<td data-label=\"Key Features\">Composition/structure featurization</td>\n</tr>\n<tr>\n<td data-label=\"Tool\">Matminer</td>\n<td data-label=\"Role\">Feature extraction</td>\n<td data-label=\"Key Features\">Magpie, SOAP, structure featurizers</td>\n</tr>\n<tr>\n<td data-label=\"Tool\">AiiDA/FireWorks</td>\n<td data-label=\"Role\">Workflow orchestration</td>\n<td data-label=\"Key Features\">Provenance, distributed execution</td>\n</tr>\n</tbody></table></div><h4 id=\"10-2-end-to-end-workflow-design\">10.2 End-to-End Workflow Design</h4><pre><code>Input: Material m with descriptors phi(m)\n1. Predict: mu_k(m), sigma_k(m) for all k potentials via GP surrogate\n2. Select: (m, k*) = argmax acquisition(mu, sigma, cost)\n3. Evaluate: Run potential k* on material m, observe y\n4. Update: Incorporate (m, k*, y) into GP posterior\n5. Recommend: Return best predicted potential or BMA prediction\n</code></pre>\n<h4 id=\"10-3-validation-and-benchmarking-protocols\">10.3 Validation and Benchmarking Protocols</h4><ul>\n<li>Leave-One-Material-Out Cross-Validation</li>\n<li>Time-Based Splits for Temporal Generalization</li>\n<li>Adversarial Testing for Robustness Assessment</li>\n</ul>\n<h4 id=\"10-4-future-directions\">10.4 Future Directions</h4><ul>\n<li>Foundation Models for Universal Potential Pre-Training</li>\n<li>Causal Inference for Mechanistic Understanding</li>\n<li>Federated Learning for Distributed Potential Development</li>\n</ul>\n<hr>\n<h2 id=\"key-findings-for-glim\">Key Findings for GLIM</h2><ol>\n<li><strong>Ensemble variance from 23 potentials is the most promising acquisition criterion</strong> for active learning</li>\n<li><strong>Heteroscedastic GP surrogates</strong> with Matern-5/2 kernels provide optimal balance for error prediction</li>\n<li><strong>BMA with property-dependent weights</strong> captures the specialist nature of potentials</li>\n<li><strong>Multi-fidelity optimization</strong> exploits the cost hierarchy via alpha-UA acquisition functions</li>\n<li><strong>Committee variance decomposes</strong> into bias + epistemic + aleatoric components</li>\n<li><strong>Leave-one-material-out cross-validation</strong> is the primary validation protocol</li>\n</ol>\n<hr>\n<p><em>Report extracted from Kimi Deep Research. March 28, 2026.</em>\n<em>Note: Partial extraction (~40% of 55K original) due to content filtering. Full report accessible in browser tab 1899282409.</em></p>\n"}