← Search

Yibo Yang

40 accepted papers

2026

A Guardrail for Safety Preservation: When Safety-Sensitive Subspace Meets Harmful-Resistant Null-Space

ICLR 2026poster

Large language models (LLMs) have achieved remarkable success in diverse tasks, yet their safety alignment remains fragile during adaptation. Even when fine-tuning on benign data or with low-rank adaptation, pre-trained safety behaviors are easily degraded, leading to harmful responses in the fine-…

Cited by 0SourceScholar
2026

CARE: Covariance-Aware and Rank-Enhanced Decomposition for Enabling Multi-Head Latent Attention

ICLR 2026poster

Converting pretrained attention modules such as *grouped-query attention* (GQA) into *multi-head latent attention* (MLA) can improve expressivity without increasing KV-cache cost, making it attractive for efficient inference. However, existing conversion methods typically apply naïve singular value…

Cited by 0SourceScholar
2026

Imitating the Truth: Attention-aware Truth-Guided Enhancement for Hallucination Mitigation in Large Vision-Language Models

ICLR 2026poster

Large Vision-Language Models (LVLMs) achieve impressive multimodal reasoning but remain prone to hallucinations, generating content inconsistent with visual evidence. Existing mitigation methods often rely on auxiliary modules or coarse decoding-time adjustments, overlooking the fine-grained dynamic…

Cited by 0SourceScholar
2025

AstroCompress: A benchmark dataset for multi-purpose compression of astronomical data

ICLR 2025poster

The site conditions that make astronomical observatories in space and on the ground so desirable---cold and dark---demand a physical remoteness that leads to limited data transmission capabilities. Such transmission limitations directly bottleneck the amount of data acquired and in an era of costly…

2025

Enhancing Online Continual Learning with Plug-and-Play State Space Model and Class-Conditional Mixture of Discretization

CVPR 2025poster

Online continual learning (OCL) seeks to learn new tasks from data streams that appear only once, while retaining knowledge of previously learned tasks. Most existing methods rely on replay, focusing on enhancing memory retention through regularization or distillation. However, they often overlook t…

2025

Inference Compute-Optimal Video Vision Language Models

ACL 2025long

This work investigates the optimal allocation of inference compute across three key scaling factors in video vision language models: language model size, frame count, and the number of visual tokens per frame. While prior works typically focuses on optimizing model efficiency or improving performanc…

Cited by 0SourcePDFScholar
2025

Instruction-Oriented Preference Alignment for Enhancing Multi-Modal Comprehension Capability of MLLMs

ICCV 2025poster

Preference alignment has emerged as an effective strategy to enhance the performance of Multimodal Large Language Models (MLLMs) following supervised fine-tuning. While existing preference alignment methods predominantly target hallucination factors, they overlook the factors essential for multi-mod…

2025

Optimization Inspired Few-Shot Adaptation for Large Language Models

NeurIPS 2025spotlight

Large Language Models (LLMs) have demonstrated remarkable performance in real-world applications. However, adapting LLMs to novel tasks via fine-tuning often requires substantial training data and computational resources that are impractical in few-shot scenarios. Existing approaches, such as In-con…

Cited by 0SourceScholar
2025

RMP-SAM: Towards Real-Time Multi-Purpose Segment Anything

ICLR 2025oral

Recent segmentation methods, which adopt large-scale data training and transformer architecture, aim to create one foundation model that can perform multiple tasks. However, most of these methods rely on heavy encoder and decoder frameworks, hindering their performance in real-time scenarios.…

2025

Transformers for Mixed-type Event Sequences

NeurIPS 2025spotlight

Event sequences appear widely in domains such as medicine, finance, and remote sensing, yet modeling them is challenging due to their heterogeneity: sequences often contain multiple event types with diverse structures—for example, electronic health records that mix discrete events like medical proce…

Cited by 0SourcecodeScholar
2024

CorDA: Context-Oriented Decomposition Adaptation of Large Language Models for Task-Aware Parameter-Efficient Fine-tuning

NeurIPS 2024poster

Current parameter-efficient fine-tuning (PEFT) methods build adapters widely agnostic of the context of downstream task to learn, or the context of important knowledge to maintain. As a result, there is often a performance gap compared to full-parameter fine-tuning, and meanwhile the fine-tuned mode…

2024

Towards Interpretable Deep Local Learning with Successive Gradient Reconciliation

ICML 2024poster

Relieving the reliance of neural network training on a global back-propagation (BP) has emerged as a notable research topic due to the biological implausibility and huge memory consumption caused by BP. Among the existing solutions, local learning optimizes gradient-isolated modules of a neural netw…

Cited by 1SourcePDFScholar
2023

DeMT: Deformable Mixer Transformer for Multi-Task Learning of Dense Prediction

AAAI 2023technical

Convolution neural networks (CNNs) and Transformers have their own advantages and both have been widely used for dense prediction in multi-task learning (MTL). Most of the current studies on MTL solely rely on CNN or Transformer. In this work, we present a novel MTL model by combining both merits of…

2023

Estimating the Rate-Distortion Function by Wasserstein Gradient Descent

NeurIPS 2023poster

In the theory of lossy compression, the rate-distortion (R-D) function $R(D)$ describes how much a data source can be compressed (in bit-rate) at any given level of fidelity (distortion). Obtaining $R(D)$ for a given data source establishes the fundamental performance limit for all compression algor…

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

Understanding Imbalanced Semantic Segmentation Through Neural Collapse

CVPR 2023poster

A recent study has shown a phenomenon called neural collapse in that the within-class means of features and the classifier weight vectors converge to the vertices of a simplex equiangular tight frame at the terminal phase of training for classification. In this paper, we explore the corresponding st…

2022

ART-Point: Improving Rotation Robustness of Point Cloud Classifiers via Adversarial Rotation

CVPR 2022poster

Point cloud classifiers with rotation robustness have been widely discussed in the 3D deep learning community. Most proposed methods either use rotation invariant descriptors as inputs or try to design rotation equivariant networks. However, robust models generated by these methods have limited perf…

Cited by 20PDFcodeScholar
2022

Eliminating Gradient Conflict in Reference-Based Line-Art Colorization

ECCV 2022poster

"Reference-based line-art colorization is a challenging task in computer vision. The color, texture, and shading are rendered based on an abstract sketch, which heavily relies on the precise long-range dependency modeling between the sketch and reference. Popular techniques to bridge the cross-modal…

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

Panoptic-PartFormer: Learning a Unified Model for Panoptic Part Segmentation

ECCV 2022poster

"Panoptic Part Segmentation (PPS) aims to unify panoptic segmentation and part segmentation into one task. Previous work mainly utilizes separated approaches to handle thing, stuff, and part predictions individually without performing any shared computation and task association. In this work, we aim…

2022

PolyphonicFormer: Unified Query Learning for Depth-Aware Video Panoptic Segmentation

ECCV 2022poster

"The Depth-aware Video Panoptic Segmentation (DVPS) is a new challenging vision problem that aims to predict panoptic segmentation and depth in a video simultaneously. The previous work solves this task by extending the existing panoptic segmentation method with an extra dense depth prediction and i…

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

View Vertically: A Hierarchical Network for Trajectory Prediction via Fourier Spectrums

ECCV 2022poster

"Understanding and forecasting future trajectories of agents are critical for behavior analysis, robot navigation, autonomous cars, and other related applications. Previous methods mostly treat trajectory prediction as time sequence generation. Different from them, this work studies agents’ trajecto…

2021

Hierarchical Autoregressive Modeling for Neural Video Compression

ICLR 2021poster

Recent work by Marino et al. (2020) showed improved performance in sequential density estimation by combining masked autoregressive flows with hierarchical latent variable models. We draw a connection between such autoregressive generative models and the task of lossy video compression. Specifically…

Cited by 52SourcePDFScholar
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
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

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

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
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