Improved Training of Wasserstein GANs, Ishaan Gulrajani, Faruk Ahmed, Martin Arjovsky, Vincent Dumoulin, Aaron Courville, 2017Advances in Neural Information Processing Systems 30 (NIPS 2017)DOI: 10.48550/arXiv.1704.00028 - Presents the gradient penalty method to enforce the Lipschitz constraint in Wasserstein GANs, leading to improved training stability and performance.
Wasserstein GAN, Martin Arjovsky, Soumith Chintala, Léon Bottou, 2017Proceedings of the 34th International Conference on Machine Learning (ICML 2017)DOI: 10.48550/arXiv.1701.07875 - Introduces the theoretical framework of Wasserstein GANs, addressing the limitations of traditional GANs and laying the foundation for WGAN-GP.
torch.autograd, PyTorch Core Team, 2024 (PyTorch Foundation) - Official PyTorch documentation for its automatic differentiation engine, essential for understanding and implementing gradient calculations like the gradient penalty.