C++ Based Open Source Framework WICWIU.v2 That Supports Generative Adversarial Nets and Metric Learning
Donghwee Yoon, Yeseong Kwon, Gyeonghyeop Kim, Cham-Jin Park, Seonggyeol Yoon, Eunseo Choi, Injung Kim · 정보과학회 컴퓨팅의 실제 논문지 · 2020
본 논문은 국내 대학 최초의 오픈소스 딥러닝 프레임워크 WICWIU(위큐)가 2018년 공개된 이후 1년 동안 개선된 내용을 소개한다. 합성곱 신경망(convolution neural networks, CNN)에 중점을 둔 WICWIU.v1의 기능 외에 생성적 적대 신경망(generative adversarial networks, GAN)과 척도 학습(metric learning)을 지원하기 위한 기능들이 추가되었다. WICWIU.v2는 Vanilla GAN이나 DCGAN과 같은 기본적인 GAN 모델뿐 아니라 WGAN, BEGAN 등 다양한 고급 GAN 모델들을 구현할 수 있는 클래스와 함수들을 제공한다. 또한, 삼중항 손실 함수와 양성 및 음성 샘플링이 가능한 데이터 로더 등 척도학습을 위한 기능들은 원샷 학습, 또는 메타 학습 알고리즘의 구현에 유용하다. WICWIU.v2는 GitHub에서 다운로드 가능하며 연구 및 상용 소프트웨어 개발에 자유롭게 활용할 수 있다.