arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2605.07810physics.opticscs.CV

预训练实现非凡全光图像去噪

Pre-training Enables Extraordinary All-optical Image Denoising

Ministry of Industry and Information Technology Key Lab of Micro-Nano Optoelectronic Information System(工业和信息化部微纳光电信息系统重点实验室);Guangdong Provincial Key Laboratory of Semiconductor Optoelectronic Materials and Intelligent Photonic System(广东省半导体光电材料与智能光子系统重点实验室);Harbin Institute of Technology(哈尔滨工业大学);School of Electronics and Information Engineering(电子与信息工程学院);Zhejiang Provincial Key Laboratory of Intelligent Vehicle Electronics Research(浙江省智能车辆电子研究所);Hangzhou Dianzi University(杭州电子科技大学);School of Science and Engineering(科学与工程学院);The Chinese University of Hong Kong (Shenzhen)(香港中文大学(深圳));Quantum Science Center of Guangdong-Hong Kong-Macao Greater Bay Area(粤港澳大湾区量子科学中心);Key Laboratory of Photonic Technology for Integrated Sensing and Communication, Ministry of Education(教育部光电一体化感知与通信技术重点实验室);Guangdong University of Technology(广东工业大学)

Xudong Lv, Yuxiang Sun, Shuo Wang, Nanxing Chen, Jun Guan, Jingtian Hu

AI总结:

本文提出一种预训练驱动方法,通过两步优化过程实现高效全光图像去噪,显著提升去噪质量,适用于多种图像风格,且在噪声环境下保持细节并提高PSNR。

摘要:

光学神经网络因其在速度和能效方面的潜力而成为强大的机器学习和信息处理工具。然而,这些物理模型的训练方法相较于数字模型仍不明确,导致性能欠佳。本文报告了一种预训练驱动方法,通过大规模数据集预训练和任务特定数据集微调,实现了高效自由空间光学去噪。与传统傅里叶域滤波和直接训练的衍射网络相比,该迁移学习过程在严重噪声下去噪效果更优,PSNR从低于8dB提升至超过18dB。重要的是,相同的预训练光学网络可一致微调处理多种图像风格,包括手写数字(MNIST)、胸片(ChestMNIST)、CIFAR-10图像和人脸(CelebA)。此外,本文还展示了光学去噪器在视觉应用中的关键作用,如人脸识别、车牌识别和噪声环境下无人机定位。

英文摘要:

Optical neural networks are emerging as powerful machine learning and information processing tools because of their potential advantages in speed and energy efficiency. The training methods of these physical models, however, remain underexplored compared to their digital counterparts and are leading to suboptimal performance. This paper reports a pre-training-driven approach that leads to snapshot image denoising with substantially improved quality. We demonstrated effective free-space optical denoising by a diffractive network optimized by a two-step process including (1) pre-training using a massive dataset of 3.45 million diverse but simple images and (2) fine-tuning with the corresponding task-specific datasets. Compared to conventional Fourier-domain filtering and directly trained diffractive networks, such a transfer learning process exhibited prominent advantages for denoising images degraded by severe noise, peak signal-to-noise ratio (PSNR) below 8 dB, while preserving fine image features and improving the PSNR to above 18 dB. Importantly, the same pre-trained optical network could be consistently fine-tuned to process degraded images from highly diverse styles ranging from handwritten digits (MNIST) and chest X-rays (ChestMNIST) to CIFAR-10 images and human faces (CelebA). We further demonstrated the critical role of our optical denoisers in vision-based applications, including face detection, plate recognition, and localization of UAVs in noisy conditions.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑