STEMGym: Benchmarking Sequential Decision-Making under Dose Budgets in Autonomous Electron Microscopy

· Source: Takara TLDR - Daily AI Papers · Field: Science & Research — Engineering & Applied Sciences, Physical Sciences & Chemistry, Research Methodology & Innovation · Depth: Expert, quick

Summary

STEMGym is an open-source Gymnasium benchmark designed for autonomous scanning transmission electron microscopy (STEM), an atomic-resolution imaging modality where electron dose causes sample damage. This benchmark features 15 physics-simulated STEM worlds, covering five materials, three difficulty levels, and four characterization tasks. It evaluates performance using the Dose-Efficiency Curve area (DEC-AUC), a scalar metric representing the information-vs-dose Pareto frontier. Research using STEMGym across 33 agent configurations under realistic dose budgets revealed that the analyst (perception) pipeline is the primary factor determining dose efficiency, not the navigation strategy. For instance, pairing a trained CNN analyst with naive raster scanning increased DEC-AUC by 5.5x (0.287 vs. 0.052) compared to a CNN-free raster baseline, with no statistically significant additional gains from Bayesian or adaptive finite-state-machine navigation. Furthermore, production-tier vision-language models performed ~13x worse than task-specific CNNs for crystallographic defect analysis. STEMGym provides infrastructure to re-evaluate ML investment in autonomous electron microscopy by separating perception, navigation, and planning.

Key takeaway

For Machine Learning Engineers developing autonomous electron microscopy systems, you should prioritize investment in your perception (analyst) pipeline over complex navigation algorithms. Your efforts on a trained CNN analyst with even naive raster scanning can yield 5.5x greater dose efficiency than advanced navigation methods. Focus on task-specific CNNs; general vision-language models underperform by ~13x for crystallographic defect analysis. This ensures your resource allocation targets the highest impact area for sample preservation and data quality.

Key insights

The perception pipeline, not navigation, is the dominant factor for dose efficiency in autonomous STEM imaging.

Principles

Method

STEMGym benchmarks autonomous STEM by simulating 15 worlds, scoring agents via Dose-Efficiency Curve area (DEC-AUC) to quantify information-vs-dose trade-offs, decoupling perception and navigation.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Research Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Takara TLDR - Daily AI Papers.