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
إعادة النظر في نماذج دالة انتشار النقطة: إطار عمل موحد وتطبيق عالي الأداء
تعتبر نماذج دالة انتشار النقطة (PSF) أدوات أساسية في المجهر الموضعي. تقدم هذه الورقة إطار عمل نظري موحد لمشكلة نمذجة الانتشار الضوئي في أنظمة التصوير ذات الفتحة الرقمية العالية (NA)، مما يثبت التكافؤ بين الطرق القائمة على تحويل فورييه ثنائي الأبعاد والطرق القائمة على تكامل بيسل أحادي البعد. طورت الدراسة مكتبة مفتوحة المصدر عالية الأداء قائمة على PyTorch، مع تحقيق أول اختبار معياري منهجي للنماذج المختلفة. تظهر التجارب أن استراتيجية بيسل مثالية للحزم المحورية المتماثلة، بينما تناسب طرق فورييه السيناريوهات الأكثر عمومية. يدعم هذا العمل الحسابات الفعالة على وحدات المعالجة المركزية والرسومات، مع إمكانية التكامل السلس في عمليات التعلم العميق والتحسين.
دالة انتشار النقطة (PSF) هي مفهوم أساسي في المجهر الضوئي، تصف الاستجابة النبضية لنظام التصوير. في أنظمة التصوير ذات الفتحة الرقمية العالية (NA)، يعتبر نموذج PSF الدقيق حاسماً للتطبيقات التالية:
المجهر الموضعي للجزيئات الفردية (SMLM): يتطلب PSF دقيق لتحديد موقع الجزيئات
المجهر بإزالة الالتفاف: يحتاج إلى نموذج PSF دقيق لاستعادة الصورة
هندسة PSF: تحقيق التصوير فوق الحد الأدنى للدقة من خلال تصميم PSF خاص
على المستوى النظري: توجد طريقتان رئيسيتان (طريقة تحويل فورييه وطريقة دالة بيسل)، لكن العلاقة بينهما غير واضحة، وتفتقر إلى مقارنة منهجية
على مستوى التطبيق: تم كتابة المكتبات الموجودة بشكل أساسي بلغة Java أو MATLAB (مثل PSF Generator السابق للمؤلفين)، مما يصعب التكامل مع أطر العمل الحديثة للتعلم العميق
على مستوى التطبيق: يفتقد الاختبار المعياري المنهجي للدقة وسرعة الحساب، مما يجعل من الصعب على المستخدمين اختيار النموذج المناسب
إطار عمل نظري موحد: بدءاً من تكامل Richards-Wolf، يثبت أن طرق فورييه (الديكارتية) وبيسل (الكروية) هي في الأساس أشكال مختلفة من نفس تكامل الانتشار
عامل تصحيح عام: اشتقاق منهجي لعوامل التصحيح الفيزيائية المتعددة (انحراف Gibson-Lanni الكروي، عامل الانقراض، معامل نقل Fresnel، تشويه الطور التعسفي، إلخ)، وتطبيقها بشكل موحد على كلا الطريقتين
تطبيق PyTorch عالي الأداء: تطوير مكتبة مفتوحة المصدر psf-generator، تطبق أربعة ناشرين (عددي/متجه × ديكارتي/كروي)، يدعم حسابات CPU/GPU والتمييز التلقائي
اختبار معياري منهجي: أول مقارنة شاملة للدقة وسرعة الحساب لنماذج PSF المختلفة، توفير إرشادات الاختيار للتطبيقات العملية
تكامل النظام البيئي: توفير واجهة رسومية لـ napari وتكامل إطار عمل chromatix البصري، لتعزيز اعتماد المجتمع مفتوح المصدر
التوحيد النظري: بدءاً من تكامل Richards-Wolf، يثبت أن طرق فورييه (ديكارتية) وبيسل (كروية) هي معاملات مختلفة لنفس تكامل الانتشار، مما يزيل الغموض النظري طويل الأمد
اختيار الطريقة:
حدقة محورية متماثلة: طريقة كروية (دقة عالية، قابلية توسع GPU)
تشويه غير متماثل: طريقة ديكارتية (عمومية قوية)
التوصية العامة: طريقة ديكارتية متجهة (توازن الأداء والتطبيق)
مزايا الأداء: تحسن دقة النموذج المتجه يتجاوز بكثير تكلفة الحساب، يجب إعطاء الأولوية للاستخدام في أنظمة NA العالية
المساهمة مفتوحة المصدر: توفير مكتبة PyTorch كاملة ومكون napari وتكامل chromatix، يدعم حسابات CPU/GPU والتمييز التلقائي
Kirshner et al. (2013): إصدار Java السابق من PSF Generator للمؤلفين
Miora et al. (2024): أحدث مراجعة شاملة لطرق حساب PSF وتطبيق MATLAB
التقييم الشامل: هذه ورقة ممتازة تجمع بين النظرية والممارسة بشكل وثيق. من الناحية النظرية، أول إثبات صارم لتكافؤ الطريقتين الرئيسيتين، إنشاء إطار عمل موحد؛ من الناحية العملية، توفير تطبيق PyTorch عالي الجودة واختبار معياري منهجي. روح المصدر المفتوح والتكامل النظام البيئي (napari و chromatix) تعكس مساهمة مهمة لمجتمع البحث. أوجه القصور الرئيسية تكمن في نقص المقارنة المباشرة مع البيانات التجريبية والبرامج الموجودة. سيصبح هذا العمل أداة مهمة في مجال التصوير المجهري الحسابي، خاصة في التطبيقات التي تجمع بين التعلم العميق والنماذج الفيزيائية.