Deep Learning

AE(Auto Encoder), VAE(Variational Auto Encoder)이란?

씨주 2024. 12. 22. 20:08

AE(Auto Encoder)

https://kr.mathworks.com/discovery/autoencoder.html

input 데이터를 latent space로 압축한 후 원본 데이터와 유사하게 reconstruction한 output 데이터 출력

  • Encoder: 입력 데이터를 저차원(latent space)으로 압축
  • Latent Space: 중요한 특징만을 보존한 저차원 표현
  • Decoder: Latent Space에서 원래 입력과 유사한 데이터를 복원

 

VAE(Variational Auto Encoder)

https://medium.com/@jwbtmf/comparison-of-autoencoders-vs-variational-autoencoders-7993442bb377

input 데이터를 확률 분포로 변환해 더 일반화된 latent space를 학습하고, 이를 샘플링하여 새로운 데이터 생성 가능

KL Divergence를 이용해 정규분포를 따르게 하여 일반화가 잘 되도록 한다.

덕분에 새로운 샘플을 뽑더라도 자연스럽고 의미있는 데이터를 추출할 수 있다.