PAM-UNet은 인코더 암과 디코더 암으로 구성된 U자 형태의 아키텍처를 가지고 있다. 인코더 암에는 모바일 컨볼루션 블록과 인버티드 잔차 블록, 차원 축소 컨볼루션 레이어가 포함되어 있다. 디코더 암은 인코더 암의 구조를 거울링하며, 스킵 연결에 점진적 루옹 어텐션 메커니즘을 적용한다. 이를 통해 관련 정보에 주목하면서도 효율성을 유지할 수 있다. 최종 레이어에서는 전치 컨볼루션을 통해 분할 마스크를 생성한다.
점진적 루옹 어텐션 메커니즘은 다중 레이어의 정보를 점진적으로 집계하여 계층적 문맥 의존성을 포착한다. 이는 단일 레이어의 정보만을 활용하는 표준 루옹 어텐션보다 우수한 성능을 보인다. 또한 어텐션 정규화를 통해 과도한 어텐션 집중을 방지한다.
실험 결과, PAM-UNet은 LiTS-2017과 Kvasir-SEG 데이터셋에서 기존 모델들을 능가하는 분할 성능을 보였다. 특히 모바일 백본 기반 모델인 MobileUNet과 비교했을 때 큰 성능 향상을 달성하면서도 계산량은 크게 줄였다. 정성적 평가에서도 PAM-UNet이 관심 영역을 정확하게 분할하는 것을 확인할 수 있다.
To Another Language
from source content
arxiv.org
Key Insights Distilled From
by Abhijit Das,... at arxiv.org 05-03-2024
https://arxiv.org/pdf/2405.01503.pdfDeeper Inquiries