UNet本质是求解控制问题的一步分裂算法
A Mathematical Explanation of UNet
- 用多网格分解控制变量，通过算子分裂求解
- 数学推导证明UNet结构恰好对应一步分裂算法
- 为UNet提供可解释性，适合研究模型原理者
UNet架构已彻底改变图像分割领域。其通用性和高精度推动了在依赖机器学习的图像任务中的广泛应用，显著提升了相关领域的发展。本文给出一个清晰简洁的数学解释，阐明UNet各组件的含义与功能。我们证明，UNet本质上是在解决一个控制问题。通过多网格方法对控制变量进行分解，并采用算子分裂技术求解该问题，其架构恰好还原了UNet的结构。结果表明，UNet是一个针对控制问题的一步算子分裂算法。
原文摘要 · Abstract (English)
The UNet architecture has transformed image segmentation. UNet's versatility and accuracy have driven its widespread adoption, significantly advancing fields reliant on machine learning problems with images. In this work, we give a clear and concise mathematical explanation of UNet. We explain what is the meaning and function of each of the components of UNet. We will show that UNet is solving a control problem. We decompose the control variables using multigrid methods. Then, operator-splitting techniques is used to solve the problem, whose architecture exactly recovers the UNet architecture. Our result shows that UNet is a one-step operator-splitting algorithm for the control problem.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品；中文卡片由大模型生成，请以原文为准。