← Search

Jiayi Shen

19 accepted papers

2026

Distributional Vision-Language Alignment by Cauchy-Schwarz Divergence

ICLR 2026poster

Vision-language alignment is crucial for various downstream tasks such as cross-modal generation and retrieval. Previous multimodal approaches like CLIP utilize InfoNCE to maximize mutual information, primarily aligning pairwise samples across modalities while overlooking distributional differences.…

Cited by 0SourceScholar
2025

DynaPrompt: Dynamic Test-Time Prompt Tuning

ICLR 2025poster

Test-time prompt tuning enhances zero-shot generalization of vision-language models but tends to ignore the relatedness among test samples during inference. Online test-time prompt tuning provides a simple way to leverage the information in previous test samples, albeit with the risk of prompt colla…

Cited by 0SourcePDFScholar
2025

LongTail-Swap: benchmarking language models’ abilities on rare words

EMNLP 2025

Children learn to speak with a low amount of data and can be taught new words on a few-shot basis, making them particularly data-efficient learners. The BabyLM challenge aims at exploring language model (LM) training in the low-data regime but uses metrics that concentrate on the head of the word di

2025

Probabilistic Interactive 3D Segmentation with Hierarchical Neural Processes

ICML 2025poster

Interactive 3D segmentation has emerged as a promising solution for generating accurate object masks in complex 3D scenes by incorporating user-provided clicks. However, two critical challenges remain underexplored: (1) effectively generalizing from sparse user clicks to produce accurate segmentatio…

Cited by 0SourcePDFScholar
2025

Probabilistic Prototype Calibration of Vision-language Models for Generalized Few-shot Semantic Segmentation

ICCV 2025poster

Generalized Few-Shot Semantic Segmentation (GFSS) aims to extend a segmentation model to novel classes with only a few annotated examples while maintaining performance on base classes. Recently, pretrained vision-language models (VLMs) such as CLIP have been leveraged in GFSS to improve generalizati…

2024

Any-Shift Prompting for Generalization over Distributions

CVPR 2024poster

Image-language models with prompt learning have shown remarkable advances in numerous downstream vision tasks. Nevertheless conventional prompt learning methods overfit the training distribution and lose the generalization ability on the test distributions. To improve the generalization across vario…

Cited by 17SourcePDFScholar
2024

GO4Align: Group Optimization for Multi-Task Alignment

NeurIPS 2024poster

This paper proposes **GO4Align**, a multi-task optimization approach that tackles task imbalance by explicitly aligning the optimization across tasks. To achieve this, we design an adaptive group risk minimization strategy, comprising two techniques in implementation: (i) dynamical group assignment,…

2023

Episodic Multi-Task Learning with Heterogeneous Neural Processes

NeurIPS 2023spotlight

This paper focuses on the data-insufficiency problem in multi-task learning within an episodic training setup. Specifically, we explore the potential of heterogeneous information across tasks and meta-knowledge among episodes to effectively tackle each task with limited data. Existing meta-learning…

2023

SuperDisco: Super-Class Discovery Improves Visual Recognition for the Long-Tail

CVPR 2023poster

Modern image classifiers perform well on populated classes while degrading considerably on tail classes with only a few instances. Humans, by contrast, effortlessly handle the long-tailed recognition challenge, since they can learn the tail representation based on different levels of semantic abstra…

Cited by 18SourcePDFScholar
2022

Association Graph Learning for Multi-Task Classification with Category Shifts

NeurIPS 2022accept

In this paper, we focus on multi-task classification, where related classification tasks share the same label space and are learned simultaneously. In particular, we tackle a new setting, which is more realistic than currently addressed in the literature, where categories shift from training to test…

2022

NFormer: Robust Person Re-Identification With Neighbor Transformer

CVPR 2022poster

Person re-identification aims to retrieve persons in highly varying settings across different cameras and scenarios, in which robust and discriminative representation learning is crucial. Most research considers learning representations from single images, ignoring any potential interactions between…

Cited by 175PDFcodeScholar
2022

Unified Visual Transformer Compression

ICLR 2022poster

Vision transformers (ViTs) have gained popularity recently. Even without customized image operators such as convolutions, ViTs can yield competitive performance when properly trained on massive data. However, the computational overhead of ViTs remains prohibitive, due to stacking multi-head self-att…

2021

A Bit More Bayesian: Domain-Invariant Learning with Uncertainty

ICML 2021spotlight

Domain generalization is challenging due to the domain shift and the uncertainty caused by the inaccessibility of target domain data. In this paper, we address both challenges with a probabilistic framework based on variational Bayesian inference, by incorporating uncertainty into neural network wei…

2021

Learning A Minimax Optimizer: A Pilot Study

ICLR 2021poster

Solving continuous minimax optimization is of extensive practical interest, yet notoriously unstable and difficult. This paper introduces the learning to optimize(L2O) methodology to the minimax problems for the first time and addresses its accompanying unique challenges. We first present Twin-L2O,…

2021

UMEC: Unified model and embedding compression for efficient recommendation systems

ICLR 2021poster

The recommendation system (RS) plays an important role in the content recommendation and retrieval scenarios. The core part of the system is the Ranking neural network, which is usually a bottleneck of whole system performance during online inference. In this work, we propose a unified model and em…

2021

Variational Multi-Task Learning with Gumbel-Softmax Priors

NeurIPS 2021poster

Multi-task learning aims to explore task relatedness to improve individual tasks, which is of particular significance in the challenging scenario that only limited data is available for each task. To tackle this challenge, we propose variational multi-task learning (VMTL), a general probabilistic in…

2020

Eliminating the Invariance on the Loss Landscape of Linear Autoencoders

ICML 2020poster

This paper proposes a new loss function for linear autoencoders (LAEs) and analytically identifies the structure of the associated loss surface. Optimizing the conventional Mean Square Error (MSE) loss results in a decoder matrix that spans the principal subspace of the sample covariance of the data…

Cited by 16SourcePDFScholar
2019

Relational Attention Network for Crowd Counting

ICCV 2019poster

Crowd counting is receiving rapidly growing research interests due to its potential application value in numerous real-world scenarios. However, due to various challenges such as occlusion, insufficient resolution and dynamic backgrounds, crowd counting remains an unsolved problem in computer vision…

Cited by 214PDFScholar