← Search

James Tin-Yau Kwok

5 accepted papers

2021

SparseBERT: Rethinking the Importance Analysis in Self-attention

ICML 2021spotlight

Transformer-based models are popularly used in natural language processing (NLP). Its core component, self-attention, has aroused widespread interest. To understand the self-attention mechanism, a direct method is to visualize the attention map of a pre-trained model. Based on the patterns observed,…

2020

Searching to Exploit Memorization Effect in Learning with Noisy Labels

ICML 2020poster

Sample selection approaches are popular in robust learning from noisy labels. However, how to properly control the selection process so that deep networks can benefit from the memorization effect is a hard problem. In this paper, motivated by the success of automated machine learning (AutoML), we mo…

Cited by 149SourcePDFScholar
2019

Efficient Nonconvex Regularized Tensor Completion with Structure-aware Proximal Iterations

ICML 2019oral

Nonconvex regularizers have been successfully used in low-rank matrix learning. In this paper, we extend this to the more challenging problem of low-rank tensor completion. Based on the proximal average algorithm, we develop an efficient solver that avoids expensive tensor folding and unfolding. A s…

Cited by 26SourcePDFScholar
2018

Online Convolutional Sparse Coding with Sample-Dependent Dictionary

ICML 2018oral

Convolutional sparse coding (CSC) has been popularly used for the learning of shift-invariant dictionaries in image and signal processing. However, existing methods have limited scalability. In this paper, instead of convolving with a dictionary shared by all samples, we propose the use of a sample-…

Cited by 10SourcePDFScholar