本研究開發了 ZERNIPAX,一個利用 Python 和 JAX 庫實現的 Zernike 多項式計算器。ZERNIPAX 採用 Jacobi 遞迴關係進行計算,在保持高精度的同時大幅提升了計算速度,尤其是對於高階模式數。
ZERNIPAX 的主要特點如下:
準確性: ZERNIPAX 使用 Jacobi 遞迴關係進行計算,能夠準確地計算高達 n = 100 的 Zernike 多項式,優於直接多項式求值的方法。
效率: ZERNIPAX 的 CPU 版本比其他開源實現快約 10 倍,GPU 版本對於高階模式數更是有顯著優勢。這得益於 JAX 庫的並行計算能力和高效的遞迴實現。
易用性: ZERNIPAX 是一個開源的 Python 套件,集成了 JAX 的自動微分功能,方便用於優化問題中的雅可比矩陣和海塞矩陣計算。
擴展性: ZERNIPAX 提供了針對不同輸入情況的優化函數,如單一模式集合與重複模式集合。未來還可進一步優化和擴展,以滿足更多科學和工程應用的需求。
總的來說,ZERNIPAX 是一個強大的 Zernike 多項式計算工具,能夠顯著提升計算效率和準確性,為光學系統分析、圖像處理等領域提供有力支持。
翻譯成其他語言
從原文內容
arxiv.org
深入探究