Neil C. Rabinowitz, Frank Perbet, H. Francis Song, Chiyuan Zhang, S.M. Ali Eslami, Matthew Botvinick. Machine Theory of Mind. 2017. arXiv:1802.07740

Mauricio Araya-Polo, Taylor Dahlke, Charlie Frogner, Chiyuan Zhang, Tomaso Poggio, and Detlef Hohl. Automated fault detection without seismic processing. The Leading Edge (TLE), Society of Exploration Geophysicists (SEG), 2017. PDF DOI:10.1190/tle36030208.1

Chiyuan Zhang, Samy Bengio, Moritz Hardt, Benjamin Recht, Oriol Vinyals. Understanding deep learning requires rethinking generalization. International Conference on Learning Representations (ICLR), Best Paper Award, 2017. arXiv:1611.03530 code slides poster

Tianqi Chen, Bing Xu, Chiyuan Zhang, Carlos Guestrin. Training Deep Nets with Sublinear Memory Cost. 2016. arXiv:1604.06174 code

Chiyuan Zhang*, Charlie Frogner*, Hossein Mobahi, Mauricio Araya-Polo, Tomaso Poggio. Learning with a Wasserstein Loss. Advances in Neural Information Processing Systems (NIPS), 2015. *equal contribution. PDF arXiv:1506.05439 project website MIT News

Tianqi Chen, Mu Li, Yutian Li, Min Lin, Naiyan Wang, Minjie Wang, Tianjun Xiao, Bing Xu, Chiyuan Zhang, Zheng Zhang. MXNet: A Distributed Deep Learning Framework for Efficiency and Flexibility. NIPS Workshop on LearningSys, 2015. arXiv:1512.01274 code

Xiaofei He, Chiyuan Zhang, Lijun Zhang, Xuelong Li. A-Optimal Projection for Image Representation. IEEE Trans. on Pattern Analysis and Machine Intelligence (TPAMI), 2015.

Chiyuan Zhang, Stephen Voinea, Georgios Evangelopoulos, Lorenzo Rosasco, Tomaso Poggio. Discriminative Template Learning in Group-Convolutional Networks for Invariant Speech Representations. INTERSPEECH 2015. PDF

Chiyuan Zhang, Charlie Frogner, Tomaso Poggio, Mauricio Araya-Polo, Detlef Hohl. Structured Output Learning for Automatic Geophysical Feature Detection. ICML Workshop on Features and Structures, 2015. slides

Chiyuan Zhang, Stephen Voinea, Georgios Evangelopoulos, Lorenzo Rosasco, Tomaso Poggio. Phone Classification by a Hierarchy of Invariant Representation Layers. INTERSPEECH 2014. PDF

Stephen Voinea, Chiyuan Zhang, Georgios Evangelopoulos, Lorenzo Rosasco, Tomaso Poggio. Word-level Invariant Representations from Acoustic Waveforms. INTERSPEECH, Best Student Paper, 2014. PDF

Georgios Evangelopoulos, Stephen Voinea, Chiyuan Zhang, Lorenzo Rosasco, Tomaso Poggio. Learning An Invariant Speech Representation. CBMM Memo No. 22, 2014. arXiv:1406.3884

Chiyuan Zhang, Georgios Evangelopoulos, Stephen Voinea, Lorenzo Rosasco, Tomaso Poggio. A Deep Representation for Invariance and Music Classification. IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 2014. PDF slides

Chiyuan Zhang, Charlie Frogner, Mauricio Araya-Polo, Detlef Hohl. Machine-learning Based Automated Fault Detection in Seismic Traces. Proceedings of 76th European Association of Geoscientists and Engineers Conference & Exhibition (EAGE), 2014. PDF

Binbin Lin, Xiaofei He, Chiyuan Zhang, Ming Ji. Parallel Vector Field Embedding. Journal of Machine Learning Research (JMLR), 2013. PDF code

Chiyuan Zhang, Xiaofei He. Image Compression by Learning to Minimize the Total Error. IEEE Transactions on Circuits and Systems for Video Technology (TCSVT), 2013. PDF

Binbin Lin, Sen Yang, Chiyuan Zhang, Jieping Ye, Xiaofei He. Multi-task Vector Field Learning. Advances in Neural Information Processing Systems (NIPS), 2012. PDF

Binbin Lin, Chiyuan Zhang, Xiaofei He. Semi-supervised Regression via Parallel Field Regularization. Advances in Neural Information Processing Systems (NIPS), 2011. PDF code

Xiaofei He, Ming Ji, Chiyuan Zhang. A Variance Minimization Criterion to Feature Selection using Laplacian Regularization. IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI), 2011. PDF

Deng Cai, Chiyuan Zhang, Xiaofei He. Unsupervised feature selection for multi-cluster data. Proc. of the 16th ACM SIGKDD Int. Conf. on Knowledge Discovery and Data Mining (KDD), 2010. PDF