← Search

Jialin Mao

6 accepted papers

2023

A Picture of the Space of Typical Learnable Tasks

ICML 2023poster

We develop information geometric techniques to understand the representations learned by deep networks when they are trained on different tasks using supervised, meta-, semi-supervised and contrastive learning. We shed light on the following phenomena that relate to the structure of the space of tas…

2022

Scalable and Efficient Training of Large Convolutional Neural Networks with Differential Privacy

NeurIPS 2022accept

Large convolutional neural networks (CNN) can be difficult to train in the differentially private (DP) regime, since the optimization algorithms require a computationally expensive operation, known as the per-sample gradient clipping. We propose an efficient and scalable implementation of this clipp…

2018

Loss Functions for Multiset Prediction

NeurIPS 2018poster

We study the problem of multiset prediction. The goal of multiset prediction is to train a predictor that maps an input to a multiset consisting of multiple items. Unlike existing problems in supervised learning, such as classification, ranking and sequence generation, there is no known order among…

Cited by 24SourcePDFScholar
2017

Saliency-based Sequential Image Attention with Multiset Prediction

NeurIPS 2017poster

Humans process visual scenes selectively and sequentially using attention. Central to models of human visual attention is the saliency map. We propose a hierarchical visual architecture that operates on a saliency map and uses a novel attention mechanism to sequentially focus on salient regions and…

Cited by 27SourcePDFScholar