Cải thiện GAN bằng cách sử dụng phương thức vận chuyển tối ưu

Tóm tắt

Chúng tôi trình bày Optimal Transport GAN (OT-GAN), một biến thể của mạng lưới đối nghịch sinh ra, giảm thiểu một số liệu mới đo khoảng cách giữa phân phối máy phát và phân phối dữ liệu. Số liệu này, mà chúng tôi gọi là khoảng cách năng lượng mini-batch, kết hợp vận chuyển tối ưu ở dạng nguyên thủy với khoảng cách năng lượng được xác định trong không gian đặc điểm học đối nghịch, tạo ra một hàm khoảng cách có độ phân biệt cao với các gradient mini-batch không thiên vị. Về mặt thực nghiệm, chúng tôi chứng minh OT-GAN có độ ổn định cao khi được đào tạo với các mini-batch lớn và chúng tôi trình bày các kết quả tiên tiến về một số vấn đề chuẩn mực phổ biến để tạo hình ảnh.

Xem thêm: tài khoản ChatGPT 4 chính hãng giá rẻ tại đây