FuseUNet: A Multi-Scale Feature Fusion Method for U-like Networks
Created by
Haebom
저자
Quansong He, Xiangde Min, Kaishen Wang, Tao He
개요
본 논문은 기존 U-Net의 skip connection이 다양한 스케일의 특징 간 효과적인 상호작용 부족 및 단순한 연결/합산 연산으로 인한 정보 통합의 비효율성이라는 두 가지 한계점을 지적합니다. 이를 극복하기 위해, skip connection을 이산 노드로 간주하고, 선형 다단계 방법의 원리를 활용한 적응형 상미분 방정식(ODE) 방법을 기반으로 한 새로운 다중 스케일 특징 융합 방법을 제시합니다. 이 방법은 인코더 및 디코더 아키텍처와 독립적이며, 다양한 U-Net 유사 네트워크에 적용 가능합니다. ACDC, KiTS2023, MSD 뇌 종양, ISIC2017/2018 피부 병변 분할 데이터셋 실험 결과, 특징 활용 개선, 네트워크 파라미터 감소, 높은 성능 유지가 확인되었습니다. 코드는 https://github.com/nayutayuki/FuseUNet 에서 공개됩니다.