Paper Reading AI Learner

Improving U-Net Confidence on TEM Image Data with L2-Regularization, Transfer Learning, and Deep Fine-Tuning

2025-07-22 17:27:33
Aiden Ochoa, Xinyuan Xu, Xing Wang

Abstract

With ever-increasing data volumes, it is essential to develop automated approaches for identifying nanoscale defects in transmission electron microscopy (TEM) images. However, compared to features in conventional photographs, nanoscale defects in TEM images exhibit far greater variation due to the complex contrast mechanisms and intricate defect structures. These challenges often result in much less labeled data and higher rates of annotation errors, posing significant obstacles to improving machine learning model performance for TEM image analysis. To address these limitations, we examined transfer learning by leveraging large, pre-trained models used for natural images. We demonstrated that by using the pre-trained encoder and L2-regularization, semantically complex features are ignored in favor of simpler, more reliable cues, substantially improving the model performance. However, this improvement cannot be captured by conventional evaluation metrics such as F1-score, which can be skewed by human annotation errors treated as ground truth. Instead, we introduced novel evaluation metrics that are independent of the annotation accuracy. Using grain boundary detection in UO2 TEM images as a case study, we found that our approach led to a 57% improvement in defect detection rate, which is a robust and holistic measure of model performance on the TEM dataset used in this work. Finally, we showed that model self-confidence is only achieved through transfer learning and fine-tuning of very deep layers.

Abstract (translated)

随着数据量的不断增加,开发自动识别透射电子显微镜(TEM)图像中纳米级缺陷的方法变得至关重要。然而,与传统照片中的特征相比,由于复杂的对比机制和复杂的缺陷结构,TEM图像中的纳米级缺陷表现出更大的变异性。这些挑战通常导致标记的数据大大减少,并且注释错误率更高,这对提高机器学习模型在TEM图像分析方面的性能构成了重大障碍。 为了解决这些问题,我们研究了迁移学习方法,通过利用大型预训练模型来处理自然图像。我们证明,在使用预训练编码器和L2正则化的情况下,复杂的语义特征会被忽略,而更简单且可靠的线索则被保留下来,从而显著提高了模型的性能。然而,这种改进无法通过传统的评估指标(如F1分数)捕捉到,因为这些传统指标可能会受到被视为真实标签的人类注释错误的影响。因此,我们引入了独立于标注准确性的新型评价指标。 以UO2 TEM图像中的晶界检测为例,我们的方法将缺陷检出率提高了57%,这是一个在用于本工作的TEM数据集上衡量模型性能的稳健且全面的标准。最后,我们证明只有通过迁移学习和对非常深层进行微调才能获得模型自信心。

URL

https://arxiv.org/abs/2507.16779

PDF

https://arxiv.org/pdf/2507.16779.pdf


Tags
3D Action Action_Localization Action_Recognition Activity Adversarial Agent Attention Autonomous Bert Boundary_Detection Caption Chat Classification CNN Compressive_Sensing Contour Contrastive_Learning Deep_Learning Denoising Detection Dialog Diffusion Drone Dynamic_Memory_Network Edge_Detection Embedding Embodied Emotion Enhancement Face Face_Detection Face_Recognition Facial_Landmark Few-Shot Gait_Recognition GAN Gaze_Estimation Gesture Gradient_Descent Handwriting Human_Parsing Image_Caption Image_Classification Image_Compression Image_Enhancement Image_Generation Image_Matting Image_Retrieval Inference Inpainting Intelligent_Chip Knowledge Knowledge_Graph Language_Model LLM Matching Medical Memory_Networks Multi_Modal Multi_Task NAS NMT Object_Detection Object_Tracking OCR Ontology Optical_Character Optical_Flow Optimization Person_Re-identification Point_Cloud Portrait_Generation Pose Pose_Estimation Prediction QA Quantitative Quantitative_Finance Quantization Re-identification Recognition Recommendation Reconstruction Regularization Reinforcement_Learning Relation Relation_Extraction Represenation Represenation_Learning Restoration Review RNN Robot Salient Scene_Classification Scene_Generation Scene_Parsing Scene_Text Segmentation Self-Supervised Semantic_Instance_Segmentation Semantic_Segmentation Semi_Global Semi_Supervised Sence_graph Sentiment Sentiment_Classification Sketch SLAM Sparse Speech Speech_Recognition Style_Transfer Summarization Super_Resolution Surveillance Survey Text_Classification Text_Generation Time_Series Tracking Transfer_Learning Transformer Unsupervised Video_Caption Video_Classification Video_Indexing Video_Prediction Video_Retrieval Visual_Relation VQA Weakly_Supervised Zero-Shot