[3줄 AutoML] AMC, 미국 작전주가 아니라 뉴럴넷 컴프레서!

2021. 7. 12. 22:28AI Paper Review/AutoML Papers

<arxiv> https://arxiv.org/pdf/1802.03494v4.pdf

1. 뉴럴넷 경량화는 당연하게도 디플로이 시 매우 중요한 것 중 하나인데, 사람이 하면 그냥 쌩 노가다에 가까운 작업이다(해봐서 안다) 그래서 이것을 자동화할 수 있다면 정말 대박인데, 그래서 이 논문이 대박이다(?)

Method Overview


2. 뉴럴넷 레이어의 임베딩을 받고 적정한 압축률을 제시하는 강화학습 에이전트(DDPG)를 이용해 자동화된 뉴럴넷 압축을 수행하게 되는데, 모델의 특성상 RNN을 사용하지 않아도 되어서 가볍다. 또한 리워드는 압축률과 정확성을 모두 잘 고려한 형태로 정의된다.

Reward Function, Error에 민감하면서도 FLOPs와 Param 또한 최적화하는 형태이다.
선행 연구와의 차별점

3. Human Expert에 비해 높은 압축률과 정확도를 취할 수 있었으며, 오버피팅 또한 관찰되지 않았다.