Saved in:
Bibliographic Details
Main Authors: Tai, Xue-Cheng, Liu, Hao, Chan, Raymond H., Li, Lingfeng
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.04434
Tags: Add Tag
No Tags, Be the first to tag this record!
Table of Contents:
  • 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.