arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2201.11333eess.IVcs.CVcs.LG

Few-shot Transfer Learning for Holographic Image Reconstruction using a Recurrent Neural Network

University of California, Los Angeles(加利福尼亚大学洛杉矶分校);California Nano Systems Institute (CNSI)(加州纳米系统研究所);David Geffen School of Medicine(大卫·格芬医学院)

Luzhe Huang, Xilin Yang, Tairan Liu, Aydogan Ozcan

英文摘要:

Deep learning-based methods in computational microscopy have been shown to be powerful but in general face some challenges due to limited generalization to new types of samples and requirements for large and diverse training data. Here, we demonstrate a few-shot transfer learning method that helps a holographic image reconstruction deep neural network rapidly generalize to new types of samples using small datasets. We pre-trained a convolutional recurrent neural network on a large dataset with diverse types of samples, which serves as the backbone model. By fixing the recurrent blocks and transferring the rest of the convolutional blocks of the pre-trained model, we reduced the number of trainable parameters by ~90% compared with standard transfer learning, while achieving equivalent generalization. We validated the effectiveness of this approach by successfully generalizing to new types of samples using small holographic datasets for training, and achieved (i) ~2.5-fold convergence speed acceleration, (ii) ~20% computation time reduction per epoch, and (iii) improved reconstruction performance over baseline network models trained from scratch. This few-shot transfer learning approach can potentially be applied in other microscopic imaging methods, helping to generalize to new types of samples without the need for extensive training time and data.

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

↑