← Search

Zhouchen Lin

112 accepted papers

2026

Beyond Rational Illusion: Behaviorally Realistic Strategic Classification

ICML 2026poster

Strategic classification studies the interaction between decision models and agents who strategically manipulate their features for favorable outcomes. Existing SC frameworks typically rely on the idealized assumption that agents are strictly rational. However, evidence from behavioral economics and…

Cited by 0SourceScholar
2026

Convergence Rate Analysis of the AdamW-Style Shampoo: Unifying One-sided and Two-Sided Preconditioning

ICML 2026poster

This paper studies the AdamW-style Shampoo optimizer, an effective implementation of the classical Shampoo that notably won the external tuning track of the AlgoPerf neural network training algorithm competition. Our analysis unifies one-sided and two-sided preconditioning and establishes the conver…

Cited by 0SourceScholar
2026

DNT: a Deeply Normalized Transformer that can be trained by Momentum SGD

ICLR 2026poster

Transformers have become the de facto backbone of modern deep learning, yet their training typically demands an advanced optimizer with adaptive learning rate like AdamW, rather than a momentum SGDW (mSGDW). Previous works show that it is mainly due to a heavy-tailed distribution of the gradients. I…

Cited by 0SourceScholar
2026

DistDF: Time-series Forecasting Needs Joint-distribution Wasserstein Alignment

ICLR 2026poster

Training time-series forecast models requires aligning the conditional distribution of model forecasts with that of the label sequence. The standard direct forecast (DF) approach seeks to minimize the conditional negative log-likelihood of the label sequence, typically estimated using the mean squa…

Cited by 0SourcecodeScholar
2026

FedAdamW: A Communication-Efficient Optimizer with Convergence and Generalization Guarantees for Federated Large Models

AAAI 2026technical

AdamW has become one of the most effective optimizers for training large-scale models. We have also observed its effectiveness in the context of federated learning (FL). However, directly applying AdamW in federated learning settings poses significant challenges: (1) due to data heterogeneity, AdamW

Cited by 0SourcePDFScholar
2026

Learning to Memorize with Attributive and Associative Memory for Online Test-Time Adaptation of Vision-Language Models

ICML 2026poster

Memory-based test-time adaptation (TTA) assigns streaming test samples into class-specific memory slots based on pseudo-labels predicted by models like CLIP, and retrieves them to facilitate subsequent predictions under distribution shift. However, this process introduces two challenges: ❶ **Each sa…

Cited by 0SourceScholar
2026

LogiConBench: Benchmarking Logical Consistencies of LLMs

ICLR 2026poster

Logical consistency, the requirement that statements remain non-contradictory under logical rules, is fundamental for trustworthy reasoning, yet current LLMs often fail to maintain it even on simple inference tasks. Existing benchmarks for LLM logical consistency are not scalable, not diverse, and n…

Cited by 0SourcecodeScholar
2026

Online Pseudo-Zeroth-Order Training of Neuromorphic Spiking Neural Networks

ICLR 2026poster

Brain-inspired neuromorphic computing with spiking neural networks (SNNs) is a promising energy-efficient computational approach. However, successfully training deep SNNs in a more biologically plausible and neuromorphic-hardware-friendly way is still challenging. Most recent methods leverage spatia…

Cited by 0SourceScholar
2026

OpenIKLR: Bridging the Reasoning Gap in Open-World Scenarios via Iterative Premise Completion

ICML 2026poster

Large Language Models (LLMs) demonstrate remarkable performance across various natural language processing tasks but struggle with complex logical reasoning, particularly in real-world settings. Existing research is largely confined to the closed-world assumption, which posits that all premises requ…

Cited by 0SourceScholar
2026

PRISM: Sequence Modeling as Parallel Residual Iteration

ICML 2026poster

Generative sequence modeling faces a fundamental tension between the expressivity of Transformers and the efficiency of linear sequence models. Existing efficient architectures are theoretically bounded by shallow, single-step linear updates, while powerful iterative methods like Test-Time Training …

Cited by 0SourceScholar
2026

Quadratic Direct Forecast for Training Multi-Step Time-Series Forecast Models

ICLR 2026poster

The design of training objective is central to training time-series forecasting models. Existing training objectives such as mean squared error mostly treat each future step as an independent, equally weighted task, which we found leading to the following two issues: (1) overlook the *label autocorr…

Cited by 0SourceScholar
2026

ReFTA: Breaking the Weight Reconstruction Bottleneck in Tensorized Parameter-Efficient Fine-Tuning

CVPR 2026

Tensor-based methods have attracted growing interest due to their ability to reduce trainable parameters and offer advantages over matrix-based approaches in parameter-efficient fine-tuning (e.g., LoRA and PiSSA), particularly in capturing inter-layer correlations. However, directly applying tensor

Cited by 0SourcecodeScholar
2026

Rethinking Personalization in Large Language Models at the Token Level

ICML 2026poster

With large language models (LLMs) now performing strongly across diverse tasks, there is growing demand for them to personalize outputs for individual users. Personalization is typically framed as an additional layer on top of a base NLP task, requiring model responses to meet user-specific needs wh…

Cited by 0SourceScholar
2026

Rethinking the Flow-based Gradual Domain Adaption: A Semi-Dual Optimal Transport Perspective

ICML 2026poster

Gradual Domain Adaption (GDA) aims to mitigate domain shift by progressively adapting models from the source domain to the target domain via intermediate domains. However, real intermediate domains are often unavailable or ineffective, necessitating the synthesis of intermediate samples. Flow-based …

Cited by 0SourceScholar
2026

Transformers with Endogenous In-Context Learning: Bias Characterization and Mitigation

ICLR 2026poster

In-context learning (ICL) enables pre-trained transformers (TFs) to perform few-shot learning across diverse tasks, fostering growing research into its underlying mechanisms. However, existing studies typically assume a causally-sufficient regime, overlooking spurious correlations and prediction bia…

Cited by 0SourceScholar
2026

Unbiased Reward Modeling from Implicit Preference

ICML 2026poster

Despite the success of reinforcement learning from human feedback (RLHF) in aligning language models, current reward modeling heavily relies on explicit preference data with high collection costs. In this work, we study implicit reward modeling---learning reward models from implicit human feedback--…

Cited by 0SourceScholar
2026

Unveiling Prior-data Fitted Networks on Causal Effect Estimation: Pre-training or Finetuning?

ICML 2026poster

Amortized causal inference via Prior-data Fitted Networks (PFNs) has emerged as a promising paradigm, enabling zero-shot estimation of causal effects without the need for dataset-specific model tuning. However, the principled effectiveness of unified pre-training across general interventional regime…

Cited by 0SourceScholar
2026

Uplift Modeling with Delayed Feedback: Identifiability and Algorithms

AAAI 2026technical

Uplift modeling has obtained significant attention, with broad applications in medicine, economics, and marketing. For example, in a push notification scenario, accurately estimating the uplift of different push frequencies on user activation and notification switch close rate is critical for balanc

Cited by 0SourcePDFScholar
2026

When Tabular Foundation Models Meet Strategic Tabular Data: A Prior Alignment Approach

ICML 2026poster

Tabular foundation models via pretrained prior-data fitted networks (PFNs) achieve remarkable generalization performance on arbitrary testing tabular data, when sample distributions are independent of the deployed classifiers, i.e., a non-strategic regime. In a variety of real-world scenarios, howev…

Cited by 0SourceScholar
2025

AdaMSS: Adaptive Multi-Subspace Approach for Parameter-Efficient Fine-Tuning

NeurIPS 2025poster

In this paper, we propose AdaMSS, an adaptive multi-subspace approach for parameter-efficient fine-tuning of large models. Unlike traditional parameter-efficient fine-tuning methods that operate within a large single subspace of the network weights, AdaMSS leverages subspace segmentation to obtain…

Cited by 0SourcecodeScholar
2025

Affine Steerable Equivariant Layer for Canonicalization of Neural Networks

ICLR 2025poster

In the field of equivariant networks, achieving affine equivariance, particularly for general group representations, has long been a challenge. In this paper, we propose the steerable EquivarLayer, a generalization of InvarLayer (Li et al., 2024), by building on the concept of equivariants beyond in…

Cited by 0SourcePDFScholar
2025

Empowering LLMs with Logical Reasoning: A Comprehensive Survey

IJCAI 2025

Large language models (LLMs) have achieved remarkable successes on various tasks. However, recent studies have found that there are still significant challenges to the logical reasoning abilities of LLMs, which can be categorized into the following two aspects: (1) Logical question answering: LLMs o

Cited by 0SourcePDFScholar
2025

Improving Model Representation and Reducing KV Cache via Skip Connections with First Value Heads

NeurIPS 2025poster

Transformer models have driven breakthroughs across various language tasks by their strong capability to learn rich contextual representations. Scaling them to improve representation, however, often demands substantial memory and compute costs, such as the Key-Value (KV) cache used during auto-regre…

Cited by 0SourcecodeScholar
2025

Inverse Methods for Missing Data Imputation

NeurIPS 2025poster

Iterative imputation is a prevalent method for completing missing data, which involves iteratively imputing each feature by treating it as a target variable and predicting its missing values using the remaining features. However, existing iterative imputation methods exhibit two critical defects: (…

Cited by 0SourcecodeScholar
2025

Language Ranker: A Lightweight Ranking framework for LLM Decoding

NeurIPS 2025poster

Conventional research on large language models (LLMs) has primarily focused on refining output distributions, while paying less attention to the decoding process that transforms these distributions into final responses. Recent advances, such as scaling the computation of inference time with reward m…

Cited by 0SourceScholar
2025

Low-Dimension-to-High-Dimension Generalization and Its Implications for Length Generalization

ICML 2025poster

Low-Dimension-to-High-Dimension (LDHD) generalization, a subset of Out-of-Distribution (OOD) generalization, involves training on a low-dimensional subspace and testing in a high-dimensional space. Assuming instances are generated from latent variables reflecting problem scale, LDHD generalization c…

Cited by 1SourcePDFScholar
2025

MME-VideoOCR: Evaluating OCR-Based Capabilities of Multimodal LLMs in Video Scenarios

NeurIPS 2025poster

Multimodal Large Language Models (MLLMs) have achieved considerable accuracy in Optical Character Recognition (OCR) from static images. However, their efficacy in video OCR is significantly diminished due to factors such as motion blur, temporal variations, and visual effects inherent in video conte…

Cited by 0SourceScholar
2025

Number Cookbook: Number Understanding of Language Models and How to Improve It

ICLR 2025poster

Large language models (LLMs) can solve an increasing number of complex reasoning tasks while making surprising mistakes in basic numerical understanding and processing (such as $9.11 > 9.9$). The latter ability is essential for tackling complex arithmetic and mathematical problems and serves as a fo…

2025

On the $O(\frac{\sqrt{d}}{K^{1/4}})$ Convergence Rate of AdamW Measured by $\ell_1$ Norm

NeurIPS 2025poster

As the default optimizer for training large language models, AdamW has achieved remarkable success in deep learning. However, its convergence behavior is not theoretically well-understood. This paper establishes the convergence rate $\frac{1}{K}\sum_{k=1}^K E[||\nabla f(x^k)||_1]\leq O(\frac{\sqrt{d…

Cited by 0SourceScholar
2025

PaZO: Preconditioned Accelerated Zeroth-Order Optimization for Fine-Tuning LLMs

NeurIPS 2025poster

This paper introduces PaZO, a preconditioned accelerated zeroth-order optimization algorithm for fine-tuning large language models (LLMs). First, we theoretically demonstrate the necessity of preconditioning in zeroth-order optimization, proving that zeroth-order stochastic gradient descent (ZO…

Cited by 0SourceScholar
2025

Projective Equivariant Networks via Second-order Fundamental Differential Invariants

NeurIPS 2025spotlight

Equivariant networks enhance model efficiency and generalization by embedding symmetry priors into their architectures. However, most existing methods, primarily based on group convolutions and steerable convolutions, face significant limitations when dealing with complex transformation groups, part…

Cited by 0SourceScholar
2025

PseuZO: Pseudo-Zeroth-Order Algorithm for Training Deep Neural Networks

NeurIPS 2025poster

Zeroth-order Optimization (ZO) has received wide attention in machine learning, especially when computing full gradient is expensive or even impossible. Recently, ZO has emerged as an important paradigm for memory-efficient fine-tuning of large language models (LLMs), circumventing the memory overhe…

Cited by 0SourcecodeScholar
2025

Pyramidal Flow Matching for Efficient Video Generative Modeling

ICLR 2025poster

Video generation requires modeling a vast spatiotemporal space, which demands significant computational resources and data usage. To reduce the complexity, the prevailing approaches employ a cascaded architecture to avoid direct training with full resolution latent. Despite reducing computational de…

2025

SEPARATE: A Simple Low-rank Projection for Gradient Compression in Modern Large-scale Model Training Process

ICLR 2025poster

Training Large Language Models (LLMs) presents a significant communication bottleneck, predominantly due to the growing scale of the gradient to communicate across multi-device clusters. However, how to mitigate communication overhead in practice remains a formidable challenge due to the weakness of…

Cited by 0SourcePDFScholar
2025

Stepsize anything: A unified learning rate schedule for budgeted-iteration training

NeurIPS 2025poster

The expanding computational costs and limited resources underscore the critical need for budgeted-iteration training, which aims to achieve optimal learning within predetermined iteration budgets. While learning rate schedules fundamentally govern the performance of different networks and tasks, par…

Cited by 0SourceScholar
2025

TC-MoE: Augmenting Mixture of Experts with Ternary Expert Choice

ICLR 2025poster

The Mixture of Experts (MoE) architecture has emerged as a promising solution to reduce computational overhead by selectively activating subsets of model parameters. The effectiveness of MoE models depends primarily on their routing mechanisms, with the widely adopted Top-K routing scheme used for a…

2025

Time-o1: Time-Series Forecasting Needs Transformed Label Alignment

NeurIPS 2025poster

Training time-series forecast models presents unique challenges in designing effective learning objectives. Existing methods predominantly utilize the temporal mean squared error, which faces two critical challenges: (1) label autocorrelation, which leads to bias from the label sequence likelihood;…

Cited by 0SourcecodeScholar
2025

Unbiased Recommender Learning from Implicit Feedback via Weakly Supervised Learning

ICML 2025poster

Implicit feedback recommendation is challenged by the missing negative feedback essential for effective model training. Existing methods often resort to negative sampling, a technique that assumes unlabeled interactions as negative samples. This assumption risks misclassifying potential positive sam…

2024

Affine Equivariant Networks Based on Differential Invariants

CVPR 2024poster

Convolutional neural networks benefit from translation equivariance achieving tremendous success. Equivariant networks further extend this property to other transformation groups. However most existing methods require discretization or sampling of groups leading to increased model sizes for larger g…

2024

Hebbian Learning based Orthogonal Projection for Continual Learning of Spiking Neural Networks

ICLR 2024poster

Neuromorphic computing with spiking neural networks is promising for energy-efficient artificial intelligence (AI) applications. However, different from humans who continually learn different tasks in a lifetime, neural network models suffer from catastrophic forgetting. How could neuronal operation…

2024

LION: Implicit Vision Prompt Tuning

AAAI 2024technical

Despite recent promising performances across a range of vision tasks, vision Transformers still have an issue of high computational costs. Recently, vision prompt learning has provided an economical solution to this problem without fine-tuning the whole large-scale model. However, the efficiency an…

Cited by 26SourcePDFScholar
2024

Relational Learning in Pre-Trained Models: A Theory from Hypergraph Recovery Perspective

ICML 2024poster

Foundation Models (FMs) have demonstrated remarkable insights into the relational dynamics of the world, leading to the crucial question: *how do these models acquire an understanding of world hybrid relations?* Traditional statistical learning, particularly for prediction problems, may overlook the…

Cited by 1SourcePDFScholar
2024

Separation and Bias of Deep Equilibrium Models on Expressivity and Learning Dynamics

NeurIPS 2024poster

The deep equilibrium model (DEQ) generalizes the conventional feedforward neural network by fixing the same weights for each layer block and extending the number of layers to infinity. This novel model directly finds the fixed points of such a forward process as features for prediction. Despite emp…

Cited by 0SourcePDFScholar
2024

Temporal Spiking Neural Networks with Synaptic Delay for Graph Reasoning

ICML 2024poster

Spiking neural networks (SNNs) are investigated as biologically inspired models of neural computation, distinguished by their computational capability and energy efficiency due to precise spiking times and sparse spikes with event-driven computation. A significant question is how SNNs can emulate hu…

2023

A Message Passing Perspective on Learning Dynamics of Contrastive Learning

ICLR 2023poster

In recent years, contrastive learning achieves impressive results on self-supervised visual representation learning, but there still lacks a rigorous understanding of its learning dynamics. In this paper, we show that if we cast a contrastive objective equivalently into the feature space, then its l…

2023

A Single-Loop Accelerated Extra-Gradient Difference Algorithm with Improved Complexity Bounds for Constrained Minimax Optimization

NeurIPS 2023oral

In this paper, we propose a novel extra-gradient difference acceleration algorithm for solving constrained nonconvex-nonconcave (NC-NC) minimax problems. In particular, we design a new extra-gradient difference step to obtain an important quasi-cocoercivity property, which plays a key role to signif…

Cited by 1SourcePDFScholar
2023

Balance, Imbalance, and Rebalance: Understanding Robust Overfitting from a Minimax Game Perspective

NeurIPS 2023poster

Adversarial Training (AT) has become arguably the state-of-the-art algorithm for extracting robust features. However, researchers recently notice that AT suffers from severe robust overfitting problems, particularly after learning rate (LR) decay. In this paper, we explain this phenomenon by viewing…

2023

Global Convergence of Over-parameterized Deep Equilibrium Models

AISTATS 2023poster

A deep equilibrium model (DEQ) is implicitly defined through an equilibrium point of an infinite-depth weight-tied model with an input-injection. Instead of infinite computations, it solves an equilibrium point directly with root-finding and computes gradients with implicit differentiation. In this…

Cited by 16SourcePDFScholar
2023

KDLGT: A Linear Graph Transformer Framework via Kernel Decomposition Approach

IJCAI 2023poster

In recent years, graph Transformers (GTs) have been demonstrated as a robust architecture for a wide range of graph learning tasks. However, the quadratic complexity of GTs limits their scalability on large-scale data, in comparison to Graph Neural Networks (GNNs). In this work, we propose the Kerne…

Cited by 6SourcePDFScholar
2023

Neural Collapse Inspired Feature-Classifier Alignment for Few-Shot Class-Incremental Learning

ICLR 2023top-25%

Few-shot class-incremental learning (FSCIL) has been a challenging problem as only a few training samples are accessible for each novel class in the new sessions. Finetuning the backbone or adjusting the classifier prototypes trained in the prior sessions would inevitably cause a misalignment betwee…

2023

Neural ePDOs: Spatially Adaptive Equivariant Partial Differential Operator Based Networks

ICLR 2023top-25%

Endowing deep learning models with symmetry priors can lead to a considerable performance improvement. As an interesting bridge between physics and deep learning, the equivariant partial differential operators (PDOs) have drawn much researchers' attention recently. However, to ensure the PDOs transl…

Cited by 8SourcePDFScholar
2023

Task-Robust Pre-Training for Worst-Case Downstream Adaptation

NeurIPS 2023poster

Pre-training has achieved remarkable success when transferred to downstream tasks. In machine learning, we care about not only the good performance of a model but also its behavior under reasonable shifts of condition. The same philosophy holds when pre-training a foundation model. However, the fou…

Cited by 0SourcePDFScholar
2023

Towards Memory- and Time-Efficient Backpropagation for Training Spiking Neural Networks

ICCV 2023poster

Spiking Neural Networks (SNNs) are promising energy-efficient models for neuromorphic computing. For training the non-differentiable SNN models, the backpropagation through time (BPTT) with surrogate gradients (SG) method has achieved high performance. However, this method suffers from considerable…

Cited by 67PDFcodeScholar
2023

Unbiased Stochastic Proximal Solver for Graph Neural Networks with Equilibrium States

ICLR 2023poster

Graph Neural Networks (GNNs) are widely used deep learning models that can extract meaningful representations from graph datasets and achieve great success in many machine learning tasks. Among them, graph neural networks with iterative iterations like unfolded GNNs and implicit GNNs can effectively…

Cited by 7SourcePDFScholar
2022

A Unified Contrastive Energy-based Model for Understanding the Generative Ability of Adversarial Training

ICLR 2022poster

Adversarial Training (AT) is known as an effective approach to enhance the robustness of deep neural networks. Recently researchers notice that robust models with AT have good generative ability and can synthesize realistic images, while the reason behind it is yet under-explored. In this paper, we…

Cited by 13SourcePDFScholar
2022

Chaos is a Ladder: A New Theoretical Understanding of Contrastive Learning via Augmentation Overlap

ICLR 2022poster

Recently, contrastive learning has risen to be a promising approach for large-scale self-supervised learning. However, theoretical understanding of how it works is still unclear. In this paper, we propose a new guarantee on the downstream performance without resorting to the conditional independence…

2022

G$^2$CN: Graph Gaussian Convolution Networks with Concentrated Graph Filters

ICML 2022spotlight

Recently, linear GCNs have shown competitive performance against non-linear ones with less computation cost, and the key lies in their propagation layers. Spectral analysis has been widely adopted in designing and analyzing existing graph propagations. Nevertheless, we notice that existing spectral…

Cited by 24SourcePDFScholar
2022

Inducing Neural Collapse in Imbalanced Learning: Do We Really Need a Learnable Classifier at the End of Deep Neural Network?

NeurIPS 2022accept

Modern deep neural networks for classification usually jointly learn a backbone for representation and a linear classifier to output the logit of each class. A recent study has shown a phenomenon called neural collapse that the within-class means of features and the classifier vectors converge to th…

2022

Kill a Bird with Two Stones: Closing the Convergence Gaps in Non-Strongly Convex Optimization by Directly Accelerated SVRG with Double Compensation and Snapshots

ICML 2022spotlight

Recently, some accelerated stochastic variance reduction algorithms such as Katyusha and ASVRG-ADMM achieve faster convergence than non-accelerated methods such as SVRG and SVRG-ADMM. However, there are still some gaps between the oracle complexities and their lower bounds. To fill in these gaps, th…

Cited by 2SourcePDFScholar
2022

Online Training Through Time for Spiking Neural Networks

NeurIPS 2022accept

Spiking neural networks (SNNs) are promising brain-inspired energy-efficient models. Recent progress in training methods has enabled successful deep SNNs on large-scale tasks with low latency. Particularly, backpropagation through time (BPTT) with surrogate gradients (SG) is popularly used to enable…

2022

Optimization-Induced Graph Implicit Nonlinear Diffusion

ICML 2022spotlight

Due to the over-smoothing issue, most existing graph neural networks can only capture limited dependencies with their inherently finite aggregation layers. To overcome this limitation, we propose a new kind of graph convolution, called Graph Implicit Nonlinear Diffusion (GIND), which implicitly has…

2022

PDO-s3DCNNs: Partial Differential Operator Based Steerable 3D CNNs

ICML 2022spotlight

Steerable models can provide very general and flexible equivariance by formulating equivariance requirements in the language of representation theory and feature fields, which has been recognized to be effective for many vision tasks. However, deriving steerable models for 3D rotations is much more…

2022

Restarted Nonconvex Accelerated Gradient Descent: No More Polylogarithmic Factor in the $O(ε^-7/4)$ Complexity

ICML 2022spotlight

This paper studies the accelerated gradient descent for general nonconvex problems under the gradient Lipschitz and Hessian Lipschitz assumptions. We establish that a simple restarted accelerated gradient descent (AGD) finds an $\epsilon$-approximate first-order stationary point in $O(\epsilon^{-7/4…

2022

Rethinking Knowledge Graph Evaluation Under the Open-World Assumption

NeurIPS 2022accept

Most knowledge graphs (KGs) are incomplete, which motivates one important research topic on automatically complementing knowledge graphs. However, evaluation of knowledge graph completion (KGC) models often ignores the incompleteness---facts in the test set are ranked against all unknown triplets wh…

2022

Towards Theoretically Inspired Neural Initialization Optimization

NeurIPS 2022accept

Automated machine learning has been widely explored to reduce human efforts in designing neural architectures and looking for proper hyperparameters. In the domain of neural initialization, however, similar automated techniques have rarely been studied. Most existing initialization methods are handc…

2022

Training High-Performance Low-Latency Spiking Neural Networks by Differentiation on Spike Representation

CVPR 2022poster

Spiking Neural Network (SNN) is a promising energy-efficient AI model when implemented on neuromorphic hardware. However, it is a challenge to efficiently train SNNs due to their non-differentiability. Most existing methods either suffer from high latency (i.e., long simulation time steps), or canno…

Cited by 184PDFcodeScholar
2021

Dissecting the Diffusion Process in Linear Graph Convolutional Networks

NeurIPS 2021poster

Graph Convolutional Networks (GCNs) have attracted more and more attentions in recent years. A typical GCN layer consists of a linear feature propagation step and a nonlinear transformation step. Recent works show that a linear GCN can achieve comparable performance to the original non-linear GCN wh…

2021

Efficient Equivariant Network

NeurIPS 2021poster

Convolutional neural networks (CNNs) have dominated the field of Computer Vision and achieved great success due to their built-in translation equivariance. Group equivariant CNNs (G-CNNs) that incorporate more equivariance can significantly improve the performance of conventional CNNs. However, G-CN…

2021

GBHT: Gradient Boosting Histogram Transform for Density Estimation

ICML 2021spotlight

In this paper, we propose a density estimation algorithm called \textit{Gradient Boosting Histogram Transform} (GBHT), where we adopt the \textit{Negative Log Likelihood} as the loss function to make the boosting procedure available for the unsupervised tasks. From a learning theory viewpoint, we fi…

Cited by 16SourcePDFScholar
2021

Graph Contrastive Clustering

ICCV 2021poster

Recently, some contrastive learning methods have been proposed to simultaneously learn representations and clustering assignments, achieving significant improvements. However, these methods do not take the category information and clustering objective into consideration, thus the learned representat…

Cited by 170PDFcodeScholar
2021

Is Attention Better Than Matrix Decomposition?

ICLR 2021poster

As an essential ingredient of modern deep learning, attention mechanism, especially self-attention, plays a vital role in the global correlation discovery. However, is hand-crafted attention irreplaceable when modeling the global context? Our intriguing finding is that self-attention is not better t…

2021

Learned Extragradient ISTA with Interpretable Residual Structures for Sparse Coding

AAAI 2021technical

Recently, the study on learned iterative shrinkage thresholding algorithm (LISTA) has attracted increasing attentions. A large number of experiments as well as some theories have proved the high efficiency of LISTA for solving sparse coding problems. However, existing LISTA methods are all serial co…

Cited by 12SourcePDFScholar
2021

Leveraged Weighted Loss for Partial Label Learning

ICML 2021oral

As an important branch of weakly supervised learning, partial label learning deals with data where each instance is assigned with a set of candidate labels, whereas only one of them is true. Despite many methodology studies on learning from partial labels, there still lacks theoretical understanding…

Cited by 130SourcePDFScholar
2021

PDO-eS2CNNs: Partial Differential Operator Based Equivariant Spherical CNNs

AAAI 2021technical

Spherical signals exist in many applications, e.g., planetary data, LiDAR scans and digitalization of 3D objects, calling for models that can process spherical data effectively. It does not perform well when simply projecting spherical data into the 2D plane and then using planar convolution neural…

Cited by 16SourcePDFScholar
2021

PointFlow: Flowing Semantics Through Points for Aerial Image Segmentation

CVPR 2021poster

Aerial Image Segmentation is a particular semantic segmentation problem and has several challenging characteristics that general semantic segmentation does not have. There are two critical issues: The one is an extremely foreground-background imbalanced distribution and the other is multiple small o…

Cited by 133PDFcodeScholar
2021

Residual Relaxation for Multi-view Representation Learning

NeurIPS 2021poster

Multi-view methods learn representations by aligning multiple views of the same image and their performance largely depends on the choice of data augmentation. In this paper, we notice that some other useful augmentations, such as image rotation, are harmful for multi-view methods because they cause…

Cited by 40SourcePDFScholar
2021

Towards Improving the Consistency, Efficiency, and Flexibility of Differentiable Neural Architecture Search

CVPR 2021poster

Most differentiable neural architecture search methods construct a super-net for search and derive a target-net as its sub-graph for evaluation. There exists a significant gap between the architectures in search and evaluation. As a result, current methods suffer from an inconsistent, inefficient, a…

Cited by 57PDFScholar
2021

Training Feedback Spiking Neural Networks by Implicit Differentiation on the Equilibrium State

NeurIPS 2021spotlight

Spiking neural networks (SNNs) are brain-inspired models that enable energy-efficient implementation on neuromorphic hardware. However, the supervised training of SNNs remains a hard problem due to the discontinuity of the spiking neuron model. Most existing methods imitate the backpropagation frame…

2021

Uncertainty Principles of Encoding GANs

ICML 2021spotlight

The compelling synthesis results of Generative Adversarial Networks (GANs) demonstrate rich semantic knowledge in their latent codes. To obtain this knowledge for downstream applications, encoding GANs has been proposed to learn encoders, such that real world data can be encoded to latent codes, whi…

Cited by 8SourcePDFScholar
2020

ISTA-NAS: Efficient and Consistent Neural Architecture Search by Sparse Coding

NeurIPS 2020poster

Neural architecture search (NAS) aims to produce the optimal sparse solution from a high-dimensional space spanned by all candidate connections. Current gradient-based NAS methods commonly ignore the constraint of sparsity in the search phase, but project the optimized solution onto a sparse one by…

2020

Implicit Euler Skip Connections: Enhancing Adversarial Robustness via Numerical Stability

ICML 2020poster

Deep neural networks have achieved great success in various areas, but recent works have found that neural networks are vulnerable to adversarial attacks, which leads to a hot topic nowadays. Although many approaches have been proposed to enhance the robustness of neural networks, few of them explor…

Cited by 45SourcePDFScholar
2020

Improving Semantic Segmentation via Decoupled Body and Edge Supervision

ECCV 2020poster

the global context, or refine objects detail along their boundaries by multi-scale feature fusion. In this paper, a new paradigm for semantic segmentation is proposed. Our insight is that appealing performance of semantic segmentation requires extit{explicitly} modeling the object extit{body} and ex…

2020

Maximum-and-Concatenation Networks

ICML 2020poster

While successful in many fields, deep neural networks (DNNs) still suffer from some open problems such as bad local minima and unsatisfactory generalization performance. In this work, we propose a novel architecture called Maximum-and-Concatenation Networks (MCN) to try eliminating bad local minima…

2020

PDO-eConvs: Partial Differential Operator Based Equivariant Convolutions

ICML 2020poster

Recent research has shown that incorporating equivariance into neural network architectures is very helpful, and there have been some works investigating the equivariance of networks under group actions. However, as digital images and feature maps are on the discrete meshgrid, corresponding equivari…

2020

Spatial Pyramid Based Graph Reasoning for Semantic Segmentation

CVPR 2020poster

The convolution operation suffers from a limited receptive filed, while global modeling is fundamental to dense prediction tasks, such as semantic segmentation. In this paper, we apply graph convolution into the semantic segmentation task and propose an improved Laplacian. The graph reasoning is dir…

Cited by 227PDFScholar
2019

Deep Comprehensive Correlation Mining for Image Clustering

ICCV 2019poster

Recent developed deep unsupervised methods allow us to jointly learn representation and cluster unlabelled data. These deep clustering methods %like DAC start with mainly focus on the correlation among samples, e.g., selecting high precision pairs to gradually tune the feature representation, which…

Cited by 242PDFcodeScholar
2019

Expectation-Maximization Attention Networks for Semantic Segmentation

ICCV 2019oral

Self-attention mechanism has been widely used for various tasks. It is designed to compute the representation of each position by a weighted sum of the features at all positions. Thus, it can capture long-range relations for computer vision tasks. However, it is computationally consuming. Since the…

Cited by 783PDFScholar
2019

Self-Supervised Convolutional Subspace Clustering Network

CVPR 2019poster

Subspace clustering methods based on data self-expression have become very popular for learning from data that lie in a union of low-dimensional linear subspaces. However, the applicability of subspace clustering has been limited because practical visual data in raw form do not necessarily lie in su…

Cited by 199PDFScholar
2018

Alternating Multi-bit Quantization for Recurrent Neural Networks

ICLR 2018poster

Recurrent neural networks have achieved excellent performance in many applications. However, on portable devices with limited resources, the models are often too large to deploy. For applications on the server with large scale concurrent requests, the latency during inference can also be very critic…

Cited by 138SourcePDFScholar
2018

Convolutional Neural Networks With Alternately Updated Clique

CVPR 2018poster

Improving information flow in deep networks helps to ease the training difficulties and utilize parameters more efficiently. Here we propose a new convolutional neural network architecture with alternately updated clique (CliqueNet). In contrast to prior networks, there are both forward and backward…

2018

Joint Sub-bands Learning with Clique Structures for Wavelet Domain Super-Resolution

NeurIPS 2018poster

Convolutional neural networks (CNNs) have recently achieved great success in single-image super-resolution (SISR). However, these methods tend to produce over-smoothed outputs and miss some textural details. To solve these problems, we propose the Super-Resolution CliqueNet (SRCliqueNet) to reconst…

Cited by 85SourcePDFScholar
2018

Recurrent Squeeze-and-Excitation Context Aggregation Net for Single Image Deraining

ECCV 2018poster

Rain streaks can severely degrade the visibility, which causes many current computer vision algorithms fail to work. So it is necessary to remove the rain from images. We propose a novel deep network architecture based on deep convolutional and recurrent neural networks for single image deraining. A…

Cited by 1039SourcePDFScholar
2018

SPIDER: Near-Optimal Non-Convex Optimization via Stochastic Path-Integrated Differential Estimator

NeurIPS 2018spotlight

In this paper, we propose a new technique named \textit{Stochastic Path-Integrated Differential EstimatoR} (SPIDER), which can be used to track many deterministic quantities of interests with significantly reduced computational cost. Combining SPIDER with the method of normalized gradient descent,…

Cited by 715SourcePDFScholar
2017

Faster and Non-ergodic O(1/K) Stochastic Alternating Direction Method of Multipliers

NeurIPS 2017poster

We study stochastic convex optimization subjected to linear equality constraints. Traditional Stochastic Alternating Direction Method of Multipliers and its Nesterov's acceleration scheme can only achieve ergodic O(1/\sqrt{K}) convergence rates, where K is the number of iteration. By introducing Var…

Cited by 14SourcePDFScholar
2016

Robust Kernel Estimation With Outliers Handling for Image Deblurring

CVPR 2016poster

Estimating blur kernels from real world images is a challenging problem as the linear image formation assumption does not hold when significant outliers, such as saturated pixels and non-Gaussian noise, are present. While some existing non-blind deblurring algorithms can deal with outliers to a cert…

Cited by 130PDFScholar
2016

Tensor Robust Principal Component Analysis: Exact Recovery of Corrupted Low-Rank Tensors via Convex Optimization

CVPR 2016poster

This paper studies the Tensor Robust Principal Component (TRPCA) problem which extends the known Robust PCA to the tensor case. Our model is based on a new tensor Singular Value Decomposition (t-SVD) and its induced tensor tubal rank and tensor nuclear norm. Consider that we have a 3-way tensor X i…

Cited by 627PDFScholar
2015

A New Retraction for Accelerating the Riemannian Three-Factor Low-Rank Matrix Completion Algorithm

CVPR 2015poster

The Riemannian three-factor matrix completion (R3MC) algorithm is one of the state-of-the-art geometric optimization methods for the low-rank matrix completion problem. It is a nonlinear conjugate-gradient method optimizing on a quotient Riemannian manifold. In the line search step, R3MC approximate…

Cited by 10SourcePDFScholar
2015

Learning Semi-Supervised Representation Towards a Unified Optimization Framework for Semi-Supervised Learning

ICCV 2015poster

State of the art approaches for Semi-Supervised Learning (SSL) usually follow a two-stage framework -- constructing an affinity matrix from the data and then propagating the partial labels on this affinity matrix to infer those unknown labels. While such a two-stage framework has been successful in…

Cited by 46PDFScholar