Revisiting PSF models: unifying framework and high-performance implementation
Liu, Stergiopoulou, Chuah et al.
Localization microscopy often relies on detailed models of point spread functions. For applications such as deconvolution or PSF engineering, accurate models for light propagation in imaging systems with high numerical aperture are required. Different models have been proposed based on 2D Fourier transforms or 1D Bessel integrals. The most precise ones combine a vectorial description of the electric field and precise aberration models. However, it may be unclear which model to choose, as there is no comprehensive comparison between the Fourier and Bessel approaches yet. Moreover, many existing libraries are written in Java (e.g. our previous PSF generator software) or MATLAB, which hinders the integration into deep learning algorithms. In this work, we start from the original Richards-Wolf integral and revisit both approaches in a systematic way. We present a unifying framework in which we prove the equivalence between the Fourier and Bessel strategies and detail a variety of correction factors applicable to both of them. Then, we provide a high-performance implementation of our theoretical framework in the form of an open-source library that is built on top of PyTorch, a popular library for deep learning. It enables us to benchmark the accuracy and computational speed of different models, thus allowing for an in-depth comparison of the existing models for the first time. We show that the Bessel strategy is optimal for axisymmetric beams while the Fourier approach can be applied to more general scenarios. Our work enables efficient PSF computation on CPU or GPU, which can then be included in simulation and optimization pipelines.
academic
Überprüfung von PSF-Modellen: einheitlicher Rahmen und hochperformante Implementierung
Punktausbreitungsfunktionen (PSF) sind Kernwerkzeuge der Lokalisierungsmikroskopie. Diese Arbeit präsentiert einen einheitlichen theoretischen Rahmen für die Modellierung der Lichtausbreitung in Abbildungssystemen mit hoher numerischer Apertur (NA) und beweist die Äquivalenz von Methoden basierend auf 2D-Fourier-Transformation und 1D-Bessel-Integration. Die Forschung entwickelt eine hochperformante Open-Source-Bibliothek auf Basis von PyTorch und führt erstmals systematische Benchmarks verschiedener Modelle durch. Experimente zeigen, dass die Bessel-Strategie für achsensymmetrische Strahlen optimal ist, während die Fourier-Methode für allgemeinere Szenarien geeignet ist. Die Arbeit unterstützt effiziente CPU/GPU-Berechnungen und kann nahtlos in Deep-Learning- und Optimierungsprozesse integriert werden.
Die Punktausbreitungsfunktion (PSF) ist ein grundlegendes Konzept der optischen Mikroskopie und beschreibt die Impulsantwort eines Abbildungssystems. In Abbildungssystemen mit hoher numerischer Apertur (NA) ist ein präzises PSF-Modell für folgende Anwendungen entscheidend:
Einzelmolekül-Lokalisierungsmikroskopie (SMLM): Erfordert präzise PSF für Moleküllokalisation
Dekonvolutionsmikroskopie: Benötigt genaues PSF-Modell zur Bildwiederherstellung
PSF-Engineering: Realisierung von Superauflösungsabbildung durch Konstruktion spezieller PSF
Theoretische Ebene: Es existieren zwei Hauptmethodenklassen (Fourier-Transformations- und Bessel-Funktionsmethoden), aber ihre Beziehung ist unklar und es fehlt ein systematischer Vergleich
Implementierungsebene: Bestehende Bibliotheken sind häufig in Java oder MATLAB geschrieben (wie der PSF Generator der Autoren), was die Integration in moderne Deep-Learning-Frameworks erschwert
Anwendungsebene: Es fehlt ein systematischer Benchmark für Genauigkeit und Rechengeschwindigkeit, was es Benutzern schwer macht, geeignete Modelle auszuwählen
Theoretischer einheitlicher Rahmen: Ausgehend vom Richards-Wolf-Integral wird bewiesen, dass die Fourier- (kartesisch) und Bessel- (sphärisch) Methoden im Wesentlichen unterschiedliche Parametrisierungen desselben Ausbreitungsintegrals sind
Universelle Korrekturfaktoren: Systematische Ableitung mehrerer physikalischer Korrekturfaktoren (Gibson-Lanni-Aberration, Extinktionsfaktor, Fresnel-Transmissionskoeffizient, beliebige Phasendistorsion etc.) und deren einheitliche Anwendung auf beide Methodenklassen
Hochperformante PyTorch-Implementierung: Entwicklung der Open-Source-Bibliothek psf-generator mit vier Propagatoren (skalar/vektoriell × kartesisch/sphärisch), unterstützt CPU/GPU-Berechnung und automatische Differentiation
Systematische Benchmarks: Erstmals umfassender Vergleich verschiedener PSF-Modelle hinsichtlich Genauigkeit und Rechengeschwindigkeit, mit Richtlinien für praktische Anwendungen
Ökosystem-Integration: Bereitstellung von napari-GUI-Plugin und chromatix-Optical-Simulation-Framework-Integration zur Förderung der Akzeptanz durch die Open-Source-Community
Äquivalenzbeweis: Erstmals strenger Beweis, dass kartesische und sphärische Parametrisierung unterschiedliche Darstellungen desselben Integrals sind, wodurch langjährige theoretische Unklarheiten in der Fachdisziplin beseitigt werden
Verallgemeinerung von Korrekturfaktoren: Erweiterung von Korrekturfaktoren, die bisher nur auf die Bessel-Methode angewendet wurden (wie Gibson-Lanni, Extinktionsfaktor), auf die Fourier-Methode
Benutzerdefinierte FFT-Implementierung: Implementierung von 2D-FFT mit beliebigen Pixelgrößen basierend auf Chirp-Z-Transformation, was das Samplingproblem bei extrem kleinen Pixelgrößen in der Lokalisierungsmikroskopie löst
Effiziente numerische Integration: Sphärische Methode mit Simpson-Regel für 4. Ordnung Genauigkeit, mit Vektorisierung durch torch.vmap für Batch-Verarbeitung
Differenzierbare Bessel-Funktionen: Erweiterung der automatischen Differentiation für Bessel-Funktionen, die von PyTorch nativ nicht unterstützt werden
Diese Forschung konzentriert sich hauptsächlich auf theoretische Validierung und Algorithmus-Benchmarking ohne echte Datensätze. Testszenarien umfassen:
Analytische Lösungsvalidierung: Verwendung der Airy-Scheibe (analytische Lösung der Fourier-Transformation einer Kreisöffnung)
Theoretische Vereinigung: Ausgehend vom Richards-Wolf-Integral wird bewiesen, dass die Fourier- (kartesisch) und Bessel- (sphärisch) Methoden unterschiedliche Parametrisierungen desselben Ausbreitungsintegrals sind, wodurch langjährige theoretische Unklarheiten beseitigt werden
Universelle Empfehlung: Vektorielle kartesische Methode (ausgewogene Leistung und Anwendbarkeit)
Leistungsvorteil: Genauigkeitsverbesserung des Vektormodells übersteigt seinen Rechenaufwand deutlich, sollte in Hochna-Systemen bevorzugt werden
Open-Source-Beitrag: Bereitstellung einer vollständigen PyTorch-Bibliothek, napari-Plugin und chromatix-Integration mit Unterstützung für CPU/GPU-Berechnung und automatische Differentiation
Leutenegger et al. (2006): Schnelle Fokusfeld-Berechnung, moderne Implementierung der kartesischen Methode
Aguet (2009): Systematisches Papier zum Vektoriellen PSF-Modell (Dissertation)
Kirshner et al. (2013): Früherer PSF Generator der Autoren in Java
Miora et al. (2024): Neueste Übersicht zu PSF-Berechnungsmethoden und MATLAB-Implementierung
Gesamtbewertung: Dies ist ein ausgezeichnetes Papier mit enger Verbindung zwischen Theorie und Praxis. Theoretisch wird erstmals die Äquivalenz der beiden Hauptmethodenklassen der PSF-Modellierung bewiesen und ein einheitlicher Rahmen etabliert; praktisch wird eine hochwertige PyTorch-Implementierung und systematische Benchmarks bereitgestellt. Das Open-Source-Engagement des Papiers und die Ökosystem-Integration (napari, chromatix) zeigen bedeutende Beiträge zur Forschungscommunity. Die Hauptschwäche liegt in fehlenden direkten Vergleichen mit experimentellen Daten und bestehender Software. Diese Arbeit wird ein wichtiges Werkzeug im Bereich Computational Microscopy, besonders bei der Kombination von Deep Learning mit physikalischen Modellen, mit breiten Zukunftsaussichten.