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
Revisione dei modelli PSF: framework unificato e implementazione ad alte prestazioni
Le funzioni di diffusione puntuale (PSF) rappresentano strumenti fondamentali per la microscopia di localizzazione. Questo articolo propone un framework teorico unificato per la modellazione della propagazione ottica in sistemi di imaging ad alta apertura numerica (NA), dimostrando l'equivalenza tra metodi basati su trasformate di Fourier bidimensionali e integrali di Bessel unidimensionali. Lo studio sviluppa una libreria open-source ad alte prestazioni basata su PyTorch, implementando per la prima volta test comparativi sistematici di diversi modelli. Gli esperimenti dimostrano che la strategia di Bessel è ottimale per fasci assialmente simmetrici, mentre il metodo di Fourier è adatto a scenari più generici. Questo lavoro supporta calcolo efficiente su CPU/GPU, integrandosi perfettamente in processi di apprendimento profondo e ottimizzazione.
La funzione di diffusione puntuale (PSF) rappresenta un concetto fondamentale nella microscopia ottica, descrivendo la risposta all'impulso del sistema di imaging. Nei sistemi di imaging ad alta apertura numerica (NA), modelli PSF accurati risultano critici per le seguenti applicazioni:
Microscopia di localizzazione di singole molecole (SMLM): richiede PSF preciso per la localizzazione molecolare
Microscopia di deconvoluzione: necessita di modelli PSF accurati per il recupero dell'immagine
Ingegneria PSF: realizza imaging a super-risoluzione attraverso la progettazione di PSF speciali
La modellazione PSF costituisce il nucleo dell'imaging computazionale, influenzando direttamente le prestazioni della microscopia a super-risoluzione (come STED, MINFLUX)
I sistemi ad alta NA richiedono considerazione di effetti fisici complessi: proprietà dei campi vettoriali, aberrazione sferica, disadattamento dell'indice di rifrazione
Modelli PSF accurati migliorano la precisione di localizzazione 3D, realizzando risoluzioni spaziali che superano il limite di diffrazione ottica
Livello teorico: esistono due classi di metodi principali (metodo della trasformata di Fourier e metodo della funzione di Bessel), ma la loro relazione rimane poco chiara, mancando confronti sistematici
Livello implementativo: le librerie esistenti sono principalmente scritte in Java o MATLAB (come il precedente PSF Generator degli autori), difficili da integrare in framework moderni di apprendimento profondo
Livello applicativo: mancano test comparativi sistematici di accuratezza e velocità computazionale, rendendo difficile la selezione del modello appropriato
Framework Teorico Unificato: partendo dall'integrale di Richards-Wolf, dimostra che i metodi di Fourier (Cartesiani) e di Bessel (Sferici) rappresentano essenzialmente diverse parametrizzazioni della medesima propagazione integrale
Fattori di Correzione Universali: derivazione sistematica di molteplici fattori di correzione fisica (aberrazione sferica Gibson-Lanni, fattore di estinzione, coefficiente di trasmissione di Fresnel, distorsioni di fase arbitrarie), applicandoli uniformemente a entrambe le classi di metodi
Implementazione PyTorch ad Alte Prestazioni: sviluppo della libreria open-source psf-generator, implementando quattro propagatori (scalare/vettoriale × cartesiano/sferico), supportando calcolo su CPU/GPU e differenziazione automatica
Test Comparativi Sistematici: primo confronto completo di diversi modelli PSF in termini di accuratezza e velocità computazionale, fornendo linee guida per la selezione in applicazioni pratiche
Integrazione nell'Ecosistema: fornitura di plugin per interfaccia grafica napari e integrazione nel framework di simulazione ottica chromatix, promuovendo l'adozione nella comunità open-source
Dimostrazione di Equivalenza: prima dimostrazione rigorosa che le parametrizzazioni cartesiana e sferica rappresentano diverse espressioni dello stesso integrale, eliminando l'ambiguità teorica di lunga data nel settore
Universalizzazione dei Fattori di Correzione: generalizzazione a metodi di Fourier dei fattori di correzione precedentemente applicati solo al metodo di Bessel (come Gibson-Lanni e fattore di estinzione)
Implementazione FFT Personalizzata: realizzazione di FFT 2D per dimensioni di pixel arbitrarie basata sulla trasformata Z di chirp, risolvendo il problema di campionamento per dimensioni di pixel estremamente piccole nella microscopia di localizzazione
Integrazione Numerica Efficiente: il metodo sferico utilizza la regola di Simpson per precisione di quarto ordine, con vettorizzazione batch mediante torch.vmap
Differenziazione Automatica di Funzioni di Bessel: estensione della funzionalità di differenziazione automatica per funzioni di Bessel non nativamente supportate da PyTorch
Questo studio si concentra principalmente su verifica teorica e test comparativi di algoritmi, non coinvolgendo dataset reali. Gli scenari di test includono:
Verifica di Soluzioni Analitiche: utilizzo del disco di Airy (soluzione analitica della trasformata di Fourier di apertura circolare)
Configurazione Parametri: lunghezza d'onda 632 nm, intervallo NA 0.5-1.3, indici di rifrazione ns=1.3, ni=1.5, ng=1.5
Funzione pupillare assialmente simmetrica → metodo sferico (alta precisione + scalabilità GPU)
Distorsioni non simmetriche/ingegneria PSF → metodo cartesiano (maggiore versatilità)
Raccomandazione predefinita: metodo cartesiano vettoriale (versatilità + costo ragionevole)
Importanza degli Effetti Vettoriali: nei sistemi ad alta NA, il modello vettoriale risulta indispensabile, con l'approssimazione scalare che produce errori significativi
Impatto dei Fattori di Correzione: correzioni come l'aberrazione sferica Gibson-Lanni risultano critiche per una modellazione accurata
Complessità Computazionale:
Cartesiano: O(n log n), dove n rappresenta la dimensione del piano trasversale
Sferico: O(n), dove n rappresenta il numero di passi di integrazione
Unificazione Teorica: partendo dall'integrale di Richards-Wolf, dimostra che i metodi di Fourier (Cartesiani) e di Bessel (Sferici) rappresentano diverse parametrizzazioni della medesima propagazione integrale, eliminando l'ambiguità teorica di lunga data
Selezione del Metodo:
Funzione pupillare assialmente simmetrica: metodo sferico (alta precisione, scalabilità GPU)
Distorsioni non simmetriche: metodo cartesiano (forte versatilità)
Raccomandazione universale: metodo cartesiano vettoriale (equilibrio prestazioni-applicabilità)
Vantaggi Prestazionali: il miglioramento di precisione del modello vettoriale supera significativamente il costo computazionale, dovendo essere prioritario nei sistemi ad alta NA
Contributo Open-Source: fornitura di libreria PyTorch completa, plugin napari e integrazione chromatix, supportando calcolo CPU/GPU e differenziazione automatica
Richards & Wolf (1959): integrale originale di Richards-Wolf, fondamenti della teoria della diffrazione vettoriale
Gibson & Lanni (1991): modello di aberrazione sferica Gibson-Lanni, modellazione di mezzi stratificati
Leutenegger et al. (2006): calcolo veloce del campo focale, implementazione moderna del metodo cartesiano
Aguet (2009): articolo sistematico su modello PSF vettoriale (tesi di dottorato)
Kirshner et al. (2013): versione Java precedente di PSF Generator degli autori
Miora et al. (2024): revisione recente di metodi di calcolo PSF e implementazione MATLAB
Valutazione Complessiva: Questo rappresenta un articolo eccellente che combina strettamente teoria e pratica. Dal punto di vista teorico, dimostra per la prima volta l'equivalenza tra le due classi di metodi PSF principali, stabilendo un framework unificato; dal punto di vista pratico, fornisce un'implementazione PyTorch di alta qualità e test comparativi sistematici. Lo spirito open-source dell'articolo e l'integrazione nell'ecosistema (napari, chromatix) riflettono importanti contributi alla comunità di ricerca. La principale insufficienza risiede nella mancanza di confronti diretti con dati sperimentali e software esistente. Questo lavoro diventerà uno strumento importante nel settore dell'imaging computazionale, con ampi prospettive di applicazione specialmente nell'integrazione di apprendimento profondo e modelli fisici.