건설용 자갈 암석 종류 분류 AI 경진대회

알고리즘 | 월간 데이콘 | 비전 | 분류 | macro-f1

상금 : 데이스쿨 프로 구독권
2025.04.08 ~ 2025.05.30 09:59 + Google Calendar
773명 마감

연습

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

제출

[Private 30th] ConvNextBase + Xception Kfold 앙상블

해가울면해운대

2025.05.30 13:08 344 조회 language

안녕하세요. 먼저 좋은 아이디어 공유해주신 조치연님께 감사말씀드립니다. CNN분야는 처음이라 좋은코드를 베이스삼아 코딩했습니다.
ConvNextBase + Xception모델을 Kfold해준 뒤 최적 앙상블 가중치를 찾아주어 앙상블한 결과를 보정을 해주었습니다. 다만 보정은 어차피 결과는 정수로 나온값이라 굳이 필요없었을거같긴 합니다.

ConvNextBase모델은 전체 레이어를 두번 학습시주었고 Xception모델은 전체레이어 학습 후 파인튜닝까지 진행 해주었습니다.
더 좋은모델이나 모델별로 첫 학습부터 Kfold를 적용시켜주고 싶었으나.. 거의 하루~이틀걸리는시간과 툭하면 터지는 램때문에 거기까진 못했고 앙상블할때만 kfold교차검증을 적용해주었습니다.

정말 이번에 데이터셋도 엄청나게 크고 그만큼 시간과 GPU비용이 얼마나 나갔는지 모르겠네요..ㅎㅎ

이번대회로 많이 느낀점이 팀을 만들어야 하기 수월할 것 같다는점입니다. 혼자 앙상블까지 하려면 모델만드는데 한세월 다갈거같아요. 그래도 정말 좋은 경험 했습니다! 과적합이 너무 심했는데 다행히도 그렇게 안떨어진것도 신기하구요!ㅎㅎ

Xception = https://drive.google.com/file/d/1V0Wvw1qyBnGeA_l7uMCYHwpCutXP8Xll/view?usp=drive_link
ConvNeXtBase = https://drive.google.com/file/d/13V4Xo5ITOECTA0rVo2VGEKlMLasJ9czC/view?usp=drive_link

분석환경: 코랩pro+ A100