XLuminA: An Auto-differentiating Discovery Framework for Super-Resolution Microscopy
arXiv:2310.08408 · doi:10.1038/s41467-024-54696-y
Abstract
Driven by human ingenuity and creativity, the discovery of super-resolution techniques, which circumvent the classical diffraction limit of light, represent a leap in optical microscopy. However, the vast space encompassing all possible experimental configurations suggests that some powerful concepts and techniques might have not been discovered yet, and might never be with a human-driven direct design approach. Thus, AI-based exploration techniques could provide enormous benefit, by exploring this space in a fast, unbiased way. We introduce XLuminA, an open-source computational framework developed using JAX, which offers enhanced computational speed enabled by its accelerated linear algebra compiler (XLA), just-in-time compilation, and its seamlessly integrated automatic vectorization, auto-differentiation capabilities and GPU compatibility. Remarkably, XLuminA demonstrates a speed-up of 4 orders of magnitude compared to well-established numerical optimization methods. We showcase XLuminA's potential by re-discovering three foundational experiments in advanced microscopy. Ultimately, XLuminA identified a novel experimental blueprint featuring sub-diffraction imaging capabilities. This work constitutes an important step in AI-driven scientific discovery of new concepts in optics and advanced microscopy.
arXiv comment29 pages, 15 figures. GitHub repo available!
References in corpus (30)
- SciPy 1.0--Fundamental Algorithms for Scientific Computing in Python
- Outlook for inverse design in nanophotonics
- Electromagnetic Force and Momentum
- Electromagnetic Angular Momentum
- Nanometer resolution imaging and tracking of fluorescent molecules with minimal photon fluxes
- Expressibility and entangling capability of parameterized quantum circuits for hybrid quantum-classical algorithms
- Deep physical neural networks enabled by a backpropagation algorithm for arbitrary physical systems
- Biological measurement beyond the quantum limit
- Laguerre-Gaussian mode sorter
- Efficient and mode selective spatial mode multiplexer based on Multi-Plane Light Conversion
- The physics of optical computing
- Imaging with quantum states of light
- On scientific understanding with artificial intelligence
- DeepSTORM3D: dense three dimensional localization microscopy and point spread function design by deep learning
- Strawberry Fields: A Software Platform for Photonic Quantum Computing
- Adjoint method and inverse design for nonlinear nanophotonic devices
- Automated Search for new Quantum Experiments
- Inverse design of photonic crystals through automatic differentiation
- On-chip integrated laser-driven particle accelerator
- Miniscope3D: optimized single-shot miniature 3D fluorescence microscopy
- Light Microscopy: An ongoing contemporary revolution
- Twisted-ligth--ion interaction: the role of longitudinal fields
- Computer-inspired Quantum Experiments
- Computer-inspired concept for high-dimensional multipartite quantum gates
- A search algorithm for quantum state engineering and metrology
- Digital Discovery of 100 diverse Quantum Experiments with PyTheus
- Learning Interpretable Representations of Entanglement in Quantum Optics Experiments using Deep Generative Models
- Automated design of quantum optical experiments for device-independent quantum key distribution
- XLuminA: An Auto-differentiating Discovery Framework for Super-Resolution Microscopy
- Differentiable Microscopy Designs an All Optical Phase Retrieval Microscope