AI Challenge for Industrial Control System Threat Detection

Algorithm | Structured | Anomaly detection | Control | TaPR

  • moneyIcon Prize : Total 2,000만원
  • 2021.08.02 ~ 2021.10.22 18:00 + Google Calendar
  • 906 Users Completed

 

[Private 4th, 특별상 1st] MLP Mixer + SAM

2022.03.03 14:06 5,061 Views language

MLP Mixer 논문 마지막에서 본 다른 도메인에서도 쓰일 수 있지 않을까 하는 문장에서 출발해서 만든 모델입니다. 
Anomaly Score 에다가 이동평균을 적용해 smoothing 시켰고, 그 값은 optuna 를 통해 반복하여 결정했습니다. 
여러개의 모델을 만들어 가중 평균을 하여 최종 결과를 냈는데, 그 weight 역시 optuna 를 이용해 반복하여 결정했습니다. 
MLP 모델이 오버피팅이 쉽게 일어나서 그 해결책으로 "When Vision Transformers Outperform ResNets without Pre-training or Strong Data Augmentations" 에서 나왔던대로 Sharpness Aware Minimization 을 사용해 보았습니다. 

블로그에도 설명글을 작성하였습니다 : https://jamm-notnull.tistory.com/25

Code