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
Revisitando modelos PSF: marco unificador e implementación de alto rendimiento
La función de dispersión de puntos (PSF) es una herramienta fundamental en microscopía de localización. Este artículo propone un marco teórico unificado para modelar la propagación óptica en sistemas de imagen de alta apertura numérica (NA), demostrando la equivalencia entre métodos basados en transformadas de Fourier 2D e integrales de Bessel 1D. Se desarrolló una biblioteca de alto rendimiento basada en PyTorch con código abierto, implementando por primera vez pruebas de rendimiento sistemáticas de diferentes modelos. Los experimentos demuestran que la estrategia de Bessel es óptima para haces axialmente simétricos, mientras que el método de Fourier es más versátil. Este trabajo soporta computación eficiente en CPU/GPU e integración sin interrupciones en procesos de aprendizaje profundo y optimización.
La función de dispersión de puntos (PSF) es un concepto fundamental en microscopía óptica que describe la respuesta al impulso del sistema de imagen. En sistemas de imagen de alta apertura numérica (NA), un modelo PSF preciso es crítico para las siguientes aplicaciones:
Microscopía de localización de moléculas individuales (SMLM): requiere PSF preciso para localización molecular
Microscopía de deconvolución: necesita un modelo PSF exacto para recuperación de imágenes
Ingeniería de PSF: realiza microscopía de superresolución mediante diseño de PSF especial
El modelado de PSF es fundamental en imagen computacional, afectando directamente el rendimiento de microscopía de superresolución (como STED, MINFLUX)
Los sistemas de alta NA requieren considerar características de campos vectoriales, aberración esférica, desajuste de índice de refracción y otros efectos físicos complejos
Un modelo PSF preciso puede mejorar la precisión de localización 3D, logrando resolución espacial que supera el límite de difracción óptica
Nivel teórico: existen dos clases de métodos principales (método de transformada de Fourier y método de función de Bessel), pero su relación no es clara, careciendo de comparación sistemática
Nivel de implementación: las bibliotecas existentes se escriben principalmente en Java o MATLAB (como el PSF Generator anterior de los autores), dificultando la integración en marcos modernos de aprendizaje profundo
Nivel de aplicación: falta de pruebas sistemáticas de precisión y velocidad computacional, dificultando que los usuarios seleccionen modelos apropiados
Marco Teórico Unificado: partiendo de la integral de Richards-Wolf, demuestra que los métodos de Fourier (Cartesiano) y Bessel (Esférico) son esencialmente diferentes parametrizaciones de la misma integral de propagación
Factor de Corrección Universal: deduce sistemáticamente múltiples factores de corrección física (aberración esférica de Gibson-Lanni, factor de extinción, coeficiente de transmisión de Fresnel, distorsión de fase arbitraria, etc.), aplicándolos uniformemente a ambas clases de métodos
Implementación PyTorch de Alto Rendimiento: desarrolla la biblioteca de código abierto psf-generator, implementando cuatro propagadores (escalar/vectorial × cartesiano/esférico), soportando computación CPU/GPU y diferenciación automática
Pruebas de Rendimiento Sistemáticas: por primera vez realiza comparación integral de precisión y velocidad computacional de diferentes modelos PSF, proporcionando orientación de selección para aplicaciones prácticas
Integración del Ecosistema: proporciona complemento de interfaz gráfica napari e integración del marco de simulación óptica chromatix, promoviendo adopción por la comunidad de código abierto
Demostración de Equivalencia: primera demostración rigurosa de que la parametrización cartesiana y esférica son diferentes representaciones de la misma integral, eliminando ambigüedad teórica de larga data en el campo
Generalización de Factores de Corrección: extiende factores de corrección previamente aplicados solo al método de Bessel (como Gibson-Lanni, factor de extinción) al método de Fourier
Implementación FFT Personalizada: implementa FFT 2D de tamaño de píxel arbitrario basada en transformada Z de chirp, resolviendo problemas de muestreo de tamaños de píxel extremadamente pequeños en microscopía de localización
Integración Numérica Eficiente: el método esférico implementa precisión de 4to orden usando regla de Simpson, con procesamiento por lotes vectorizado mediante torch.vmap
Función de Bessel Diferenciable: extiende funcionalidad de diferenciación automática de función de Bessel no soportada nativamente por PyTorch
Esta investigación se enfoca principalmente en verificación teórica y pruebas de rendimiento de algoritmos, sin involucrar conjuntos de datos reales. Los escenarios de prueba incluyen:
Verificación de Solución Analítica: utilizando disco de Airy (solución analítica de transformada de Fourier de apertura circular)
Configuración de Parámetros: longitud de onda 632nm, rango NA 0.5-1.3, índices de refracción ns=1.3, ni=1.5, ng=1.5
Importancia de Efectos Vectoriales: en sistemas de alta NA, el modelo vectorial es indispensable, la aproximación escalar produce errores significativos
Impacto de Factores de Corrección: correcciones como aberración esférica de Gibson-Lanni son críticas para modelado preciso
Complejidad Computacional:
Cartesiano: O(n log n), donde n es tamaño del plano transversal
Esférico: O(n), donde n es número de pasos de integración
Unificación Teórica: partiendo de la integral de Richards-Wolf, demuestra que los métodos de Fourier (Cartesiano) y Bessel (Esférico) son diferentes parametrizaciones de la misma integral de propagación, eliminando ambigüedad teórica de larga data
Distorsión no simétrica: método cartesiano (fuerte versatilidad)
Recomendación universal: método cartesiano vectorial (equilibrio de rendimiento y aplicabilidad)
Ventaja de Rendimiento: la mejora de precisión del modelo vectorial supera significativamente el costo computacional, debería priorizarse en sistemas de alta NA
Contribución de Código Abierto: proporciona biblioteca PyTorch completa, complemento napari e integración chromatix, soportando computación CPU/GPU y diferenciación automática
Richards & Wolf (1959): integral original de Richards-Wolf, fundamento de teoría de difracción vectorial
Gibson & Lanni (1991): modelo de aberración esférica de Gibson-Lanni, modelado de medios estratificados
Leutenegger et al. (2006): cálculo rápido de campo focal, implementación moderna de método cartesiano
Aguet (2009): artículo sistemático de modelo PSF vectorial (tesis doctoral)
Kirshner et al. (2013): versión Java anterior de PSF Generator de los autores
Miora et al. (2024): revisión más reciente de métodos de cálculo PSF e implementación MATLAB
Evaluación General: Este es un artículo excelente que combina estrechamente teoría y práctica. Teóricamente, demuestra por primera vez la equivalencia de dos métodos principales de modelado PSF, estableciendo un marco unificado; prácticamente, proporciona implementación PyTorch de alta calidad y pruebas de rendimiento sistemáticas. El espíritu de código abierto del artículo e integración del ecosistema (napari, chromatix) reflejan contribución importante a la comunidad de investigación. La principal insuficiencia radica en falta de comparación directa con datos experimentales y software existente. Este trabajo se convertirá en herramienta importante en el campo de imagen microscópica computacional, con amplio potencial especialmente en aplicaciones que combinan aprendizaje profundo con modelos físicos.