Title: OmniLens++: Blind Lens Aberration Correction via Large LensLib Pre-Training and Latent PSF Representation
Primary area: applications to computer vision, audio, language, and other modalities
Keywords: Optical Aberration Correction; Universal Image Restoration; Vector Quantization; Degradation Priors Learning

Abstract: Emerging deep-learning-based lens library pre‑training (LensLib-PT) pipeline offers a new avenue for blind lens aberration correction by training a universal neural network, demonstrating strong capability in handling diverse unknown Optical Degradations (ODs).
This work proposes the OmniLens++ framework, which resolves two challenges that hinder the generalization ability of existing pipelines: the difficulty of scaling data and the absence of prior guidance characterizing OD. To improve data scalability, we expand the design specifications to increase the OD diversity of the lens source, and we sample a more uniform OD distribution by quantifying the spatial-variation patterns and severity of OD. 
In terms of model design, to leverage the Point Spread Functions (PSFs), which intuitively describe OD, as guidance in a blind paradigm, we propose the Latent PSF Representation (LPR). The VQVAE framework is introduced to learn latent features of LensLib's PSFs, which is assisted by modeling the optical degradation process to constrain the learning of OD priors. 
Experiments on diverse aberrations of real-world lenses and synthetic LensLib show that OmniLens++ exhibits state‑of‑the‑art performance in blind aberration correction. Beyond performance, the AODLibpro is verified as a scalable foundation for more effective training across diverse aberrations, and LPR can further tap the potential of large‑scale LensLib. The source code and datasets will be made publicly available.