The importance of quantifying uncertainty in deep networks has become paramount for reliable real-world applications. In this paper, we propose a method to improve uncertainty estimation in medical Image-to-Image (I2I) translation. Our model integrates aleatoric uncertainty and employs Uncertainty-Aware Regularization (UAR) inspired by simple priors to refine uncertainty estimates and enhance reconstruction quality. We show that by leveraging simple priors on parameters, our approach captures more robust uncertainty maps, effectively refining them to indicate precisely where the network encounters difficulties, while being less affected by noise. Our experiments demonstrate that UAR not only improves translation performance, but also provides better uncertainty estimations, particularly in the presence of noise and artifacts. We validate our approach using two medical imaging datasets, showcasing its effectiveness in maintaining high confidence in familiar regions while accurately identifying areas of uncertainty in novel/ambiguous scenarios.
Cascaded Multi-path Shortcut Diffusion Model for Medical Image Translation
Yale University（耶鲁大学）；University of California Irvine（加州大学欧文分校）；University of Science and Technology of China（中国科学技术大学）；Case Western Reserve University（凯斯西储大学）；Northwestern University（西北大学）；Yale School of Medicine（耶鲁医学院）
Smooth image-to-image translations with latent space interpolations
University of Trento（特伦托大学）；Fondazione Bruno Kessler（布鲁诺·凯斯勒基金会）；University of Modena and Reggio Emilia（摩德纳-雷焦艾米利亚大学）；Shanghai Jiao Tong University（上海交通大学）；Tencent AI Lab（腾讯人工智能实验室）
Double-Uncertainty Guided Spatial and Temporal Consistency Regularization Weighting for Learning-based Abdominal Registration
The Chinese University of Hong Kong（香港中文大学）；Brigham and Women’s Hospital（布莱根妇女医院）；Harvard Medical School（哈佛医学院）；Tencent Jarvis Lab（腾讯医疗健康（深圳）有限公司（腾讯医疗AI实验室/腾讯天衍实验室））；Tsinghua University（清华大学）；Shenzhen International Graduate School, Tsinghua University（清华大学深圳国际研究生院）
CURVE: Learning Causality-Inspired Invariant Representations for Robust Scene Understanding via Uncertainty-Guided Regularization
Shanghai Research Institute for Intelligent Autonomous Systems, Tongji University, Shanghai, China（上海智能自主系统研究院，同济大学）；Guohao School, Tongji University, Shanghai, China（国豪学院，同济大学）；School of Automotive Studies, Tongji University, Shanghai, China（汽车研究学院，同济大学）；College of Electronics and Information Engineering, Tongji University, Shanghai, China（电子与信息工程学院，同济大学）
Composed Image Retrieval with Text Feedback via Multi-grained Uncertainty Regularization
Sea-NExT Joint Lab, National University of Singapore（新加坡国立大学 Sea-NExT 联合实验室）；Tsinghua University（清华大学）；Faculty of Science and Technology, and Institute of Collaborative Innovation, University of Macau（澳门大学科技学院及协同创新研究院）
EmbodiedOcc++: Boosting Embodied 3D Occupancy Prediction with Plane Regularization and Uncertainty Sampler
State Key Laboratory of Multimedia Information Processing, School of Computer Science, Peking University（多媒体信息处理国家重点实验室，计算机学院，北京大学）；School of electronic science and engineering, Nanjing University（电子科学与工程学院，南京大学）；Berkeley Artificial Intelligence Research Lab, Department of EECS, University of California, Berkeley（伯克利人工智能研究实验室，电子工程与计算机科学系，加州大学伯克利分校）
Deep learning with self-supervision and uncertainty regularization to count fish in underwater images
Universitat de Barcelona（巴塞罗那大学）；Universidade Federal de Santa Catarina（圣卡塔琳娜联邦大学）；Max Planck Institute of Animal Behavior（马克斯·普朗克动物行为研究所）；Universidade Federal do Paraná（巴拉那联邦大学）；Witwatersrand University（威特沃特斯兰德大学）；Oregon State University（俄勒冈州立大学）；University of Zurich（苏黎世大学）；Universitat Autonoma de Barcelona（巴塞罗那自治大学）
