제1회 Medical AI (MAI) 경진대회

알고리즘 | 의료 | 유전자 | 비전 | 회귀 | PCC

상금 : 1,000만원
860명 마감

연습

대회안내 데이터 코드 공유 토크 리더보드

제출

ResNet model에 Attention Module(CBAM) 응용 , Regression 최적화 코드

Thyoon

2025.02.08 20:13 1,254 조회 language

ResNet model에 Attention Module(CBAM) 응용하여 채널에 관한 피처 어텐션과 spatial(위치) 어텐션 모두를 적용하여 학습성능을 최적화 하였다.
로스함수에서는 SmoothL1Loss와 HuberLoss를 결합하여, 피처가 0에 가까워져도, 피처 소실이될 확률을 최소화 하였다.
명세에 나와있는 평가지표를 구현은 하였으나 맞는지는 확실치 않아, checkpoint를 선별하는 요소에서 배제하였고, MSE를 주요 평가에 사용하였다.
피드백은 언제나 환영합니다.

코드