분석시각화 대회 코드 공유 게시물은
내용 확인 후
좋아요(투표) 가능합니다.
[Private 4위, 0.96492] | MobileNetV2(3D) + Augmentation
안녕하세요~ 데이콘은 처음으로 참가했습니다.
처음에는 "어떻게 우리가 익숙한 2차원으로 잘 투영시켜서 학습시킬까?" 라는 차원에서 접근했으나,
1) 이미 2차원 데이터에서 2D Convolution 기반 네트워크는 훌륭하게 MNIST를 잘 구분하고 있으며,
2) 3D 데이터로 구성 되었지만, 결국 MNIST 숫자 10개 예측하는 상대적으로 간단한 task 이고, 3D Convolution으로 네트워크를 만들면 잘 구분하지 않을까?
이러한 다소 직관적인 생각에서 출발했습니다.
모델의 정확도를 높이기 위해, 2D로 치면 픽셀 해상도에 해당하는 복셀 해상도를 높였습니다.(64*64*64)
또한 테스트 데이터 셋이 어떤 각도로 회전되어 있는지 모르기에 Open3D 라이브러리를 통해 X, Y, Z 축 기준으로 임의의 각도로 회전을 통해 데이터 셋을 증강했습니다.
이로 인한 연산량 증가 및 병목 현상을 완화하기 위해, 가벼운 모델인 MobileNetV2를 3D 데이터를 입력 받도록 레이어와 연산을 변환하여 사용했으며, 증강 데이터를 램에 올려서 캐싱하여 학습을 진행했습니다.
별 기대 없이 흥미로운 직관을 테스트 해보는 차원에서 참여했는데, 운좋게 상위권에 위치했던거 같습니다.
좋은 대회 열어주셔서 감사하고, 긴 기간동안 대회 참가하신 모든 분들 수고 많으셨습니다.
DACON Co.,Ltd | CEO Kookjin Kim | 699-81-01021
Mail-order-sales Registration Number: 2021-서울영등포-1704
Business Providing Employment Information Number: J1204020250004
#901, Eunhaeng-ro 3, Yeongdeungpo-gu, Seoul 07237
E-mail dacon@dacon.io |
Tel. 070-4102-0545
Copyright ⓒ DACON Inc. All rights reserved