Automated Nanoparticle Image Processing Pipeline for AI-Driven Materials Characterization
Recent innovations have made it possible to produce millions of distinct nanoparticles on a chip. These vast volumes of data are impossible to analyze manually, necessitating the development of automated tools. In previous work, we created a binary classification machine learning model to select quality nanoparticle images for downstream analysis. In this work, we show that adding a custom image preprocessing step before model training can produce significantly higher-performing models in a fraction of the time and make the model more robust to different image noise levels and microscope acquisition settings. The proposed image processing pipeline effectively cleans raw nanoparticle images, enhances key features, and allows us to use much lower resolution images and simpler neural network model architectures, resulting in higher performance and significant cost savings. Experiments demonstrate superior performance relative to our baseline, including a 15% improvement in recall and more than a 10% increase in accuracy. Given the high cost of downstream analysis, it is critical to minimize false positives in our application, and our best-performing model obtains a precision of 97.3% and weighted F-score of 95.9% on an unseen test set. Additionally, model training time is reduced from 15.5 hours to 32 seconds. We expect that adopting this pipeline for AI-driven automated nanoparticle characterization will offer a considerable speedup in the laboratory, allowing researchers to rapidly and accurately analyze much greater volumes of data and accelerate materials discovery.
Authors
- Ankit Agrawal (ORCID: https://orcid.org/0000-0002-5519-0302)
- Carolin B. Wahl (ORCID: https://orcid.org/0000-0002-1569-1030)
- Vinayak P. Dravid (ORCID: https://orcid.org/0000-0002-6007-3063)
- Wei‐keng Liao (ORCID: https://orcid.org/0009-0008-9411-2543)
- Alexandra Day (ORCID: https://orcid.org/0009-0003-7467-2504)
- Alok Choudhary (ORCID: https://orcid.org/0000-0001-8152-6319)
- Roberto dos Reis (ORCID: https://orcid.org/0000-0002-6011-6078)
Institutions
- Northwestern University (US)
Publication Details
- Published
- 2024-10-20
- DOI
- https://doi.org/10.1145/3627673.3680100
- Citations
- 4
- Primary Topic
- Machine Learning in Materials Science
- Type
- article
- Field-Weighted Citation Impact
- 0.47