🧠 模型与架构

U-Net

U-Net模型UNetU形网络语义分割U-Net

U-Net是一种对称的编码器-解码器卷积网络架构,通过跳跃连接融合浅层和深层特征,专为生物医学图像分割设计,后成为语义分割和扩散模型等领域的标准架构。


U-Net 简介

U-Net 由 Olaf Ronneberger 等人于 2015 年提出,因其对称的 U 形架构得名。U-Net 在 ISBI 细胞追踪挑战赛中大幅超越此前方法,奠定了生物医学图像分割的标杆。

架构设计

U-Net 由三部分组成:收缩路径(编码器) 使用卷积 + 池化下采样逐步提取高维特征;扩张路径(解码器) 使用转置卷积上采样恢复空间信息;跳跃连接 将编码器的浅层高分辨率特征直接传递到解码器对应层,帮助恢复空间细节。

重要影响

U-Net 的编码器-解码器 + 跳跃连接设计被广泛迁移到各类密集预测任务中。值得注意的是,U-Net 是扩散模型中最重要的组件之一——Stable Diffusion、DALL-E 2、Imagen 等文生图模型都使用 U-Net 作为骨干网络来逐步去噪生成图像。


相关术语

分享: