Reduce memory usage for ICML generation