Paper Reading AI Learner

U-Net Based Healthy 3D Brain Tissue Inpainting

2025-07-24 06:26:46
Juexin Zhang, Ying Weng, Ke Chen

Abstract

This paper introduces a novel approach to synthesize healthy 3D brain tissue from masked input images, specifically focusing on the task of 'ASNR-MICCAI BraTS Local Synthesis of Tissue via Inpainting'. Our proposed method employs a U-Net-based architecture, which is designed to effectively reconstruct the missing or corrupted regions of brain MRI scans. To enhance our model's generalization capabilities and robustness, we implement a comprehensive data augmentation strategy that involves randomly masking healthy images during training. Our model is trained on the BraTS-Local-Inpainting dataset and demonstrates the exceptional performance in recovering healthy brain tissue. The evaluation metrics employed, including Structural Similarity Index (SSIM), Peak Signal-to-Noise Ratio (PSNR), and Mean Squared Error (MSE), consistently yields impressive results. On the BraTS-Local-Inpainting validation set, our model achieved an SSIM score of 0.841, a PSNR score of 23.257, and an MSE score of 0.007. Notably, these evaluation metrics exhibit relatively low standard deviations, i.e., 0.103 for SSIM score, 4.213 for PSNR score and 0.007 for MSE score, which indicates that our model's reliability and consistency across various input scenarios. Our method also secured first place in the challenge.

Abstract (translated)

本文提出了一种新颖的方法,用于从带有遮罩的输入图像中合成健康的人脑组织,特别关注的是“ASNR-MICCAI BraTS 局部组织插值”任务。我们提出的模型采用基于U-Net架构的设计,旨在有效重建大脑MRI扫描中缺失或损坏的部分区域。为了增强模型的泛化能力和鲁棒性,我们在训练过程中实施了一种全面的数据增强策略,即在随机遮罩健康图像时进行训练。 我们的模型是在BraTS-Local-Inpainting数据集上进行训练的,并且在恢复健康的脑组织方面表现出色。评估指标包括结构相似度指数(SSIM)、峰值信噪比(PSNR)和均方误差(MSE),这些指标持续显示出色的结果。在BraTS-Local-Inpainting验证集中,我们的模型达到了0.841的SSIM分数、23.257的PSNR分数以及0.007的MSE分数。 值得注意的是,这些评估指标的标准偏差相对较低,即SSIM得分为0.103,PSNR得分为4.213,MSE得分为0.007,这表明我们的模型在各种输入场景中的可靠性和一致性。此外,我们的方法在这次挑战赛中获得了第一名的成绩。

URL

https://arxiv.org/abs/2507.18126

PDF

https://arxiv.org/pdf/2507.18126.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