ARSM: Augment-REINFORCE-Swap-Merge Estimator for Gradient Backpropagation Through Categorical Variables.

Mingzhang Yin, Yuguang Yue, Mingyuan Zhou

Browse the full ICML paper archive.