Diffusion Algorithm for Metalens Optical Aberration Correction
Harshana Weligampola, Abhiram Gnanasambandam, Weiheng Tang, Dilshan Godaliyadda, Qi Guo
Abstract
Metalenses offer a path toward creating ultra-thin optical systems, but they inherently suffer from severe, spatially varying optical aberrations, especially chromatic aberration, which makes image reconstruction a significant challenge. This paper presents a novel algorithmic solution to this problem, designed to reconstruct a sharp, full-color image from two inputs: a sharp, bandpass-filtered grayscale ``structure image'' and a heavily distorted ``color cue'' image, both captured by the metalens system. Our method utilizes a dual-branch diffusion model, built upon a pre-trained Stable Diffusion XL framework, to fuse information from the two inputs. We demonstrate through quantitative and qualitative comparisons that our approach significantly outperforms existing deblurring and pansharpening methods, effectively restoring high-frequency details while accurately colorizing the image.
BibTeX
@inproceedings{icassp2026_diffusionalgorit,
title = {Diffusion Algorithm for Metalens Optical Aberration Correction},
author = {Harshana Weligampola and Abhiram Gnanasambandam and Weiheng Tang and Dilshan Godaliyadda and Qi Guo},
booktitle = {ICASSP 2026},
year = {2026}
}