← Search

Yimeng Zhang

19 accepted papers

2026

Self-Supervised Foundation Model for Calcium-imaging Population Dynamics

ICML 2026poster

Recent work suggests that large-scale, multi-animal modeling can significantly improve neural recording analysis. However, for functional calcium traces, existing approaches remain task-specific, limiting transfer across common neuroscience objectives. To address this challenge, we propose \textbf{C…

Cited by 0SourceScholar
2026

Shop-R1: Rewarding LLMs to Simulate Human Behavior in Online Shopping via Reinforcement Learning

ICLR 2026poster

Large Language Models (LLMs) have recently demonstrated strong potential in generating ‘believable human-like’ behavior in web environments. Prior work has explored augmenting training data with LLM-synthesized rationales and applying supervised fine-tuning (SFT) to enhance reasoning ability, which…

Cited by 0SourcecodeScholar
2026

Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputs

ICLR 2026poster

Machine unlearning (MU) for large language models (LLMs), commonly referred to as LLM unlearning, seeks to remove specific undesirable data or knowledge from a trained model, while maintaining its performance on standard tasks. While unlearning plays a vital role in protecting data privacy, enforcin…

Cited by 0SourcecodeScholar
2025

Can Large Language Models Understand You Better? An MBTI Personality Detection Dataset Aligned with Population Traits

COLING 2025main

The Myers-Briggs Type Indicator (MBTI) is one of the most influential personality theories reflecting individual differences in thinking, feeling, and behaving. MBTI personality detection has garnered considerable research interest and has evolved significantly over the years. However, this task ten…

2025

Cross-Rejective Open-Set SAR Image Registration

CVPR 2025poster

Synthetic Aperture Radar (SAR) image registration is an essential upstream task in geoscience applications, in which pre-detected keypoints from two images are employed as observed objects to seek matched-point pairs. In general, the registration is regarded as a typical closed-set classification, w…

2025

ID-Patch: Robust ID Association for Group Photo Personalization

CVPR 2025poster

The ability to synthesize personalized group photos and specify the positions of each identity offers immense creative potential. While such imagery can be visually appealing, it presents significant challenges for existing technologies. A persistent issue is identity (ID) leakage, where injected fa…

2025

Rethinking Multiple-Instance Learning From Feature Space to Probability Space

ICLR 2025poster

Multiple-instance learning (MIL) was initially proposed to identify key instances within a set (bag) of instances when only one bag-level label is provided. Current deep MIL models mostly solve multi-instance problem in feature space. Nevertheless, with the increasing complexity of data, we found th…

2024

DeepZero: Scaling Up Zeroth-Order Optimization for Deep Model Training

ICLR 2024poster

Zeroth-order (ZO) optimization has become a popular technique for solving machine learning (ML) problems when first-order (FO) information is difficult or impossible to obtain. However, the scalability of ZO optimization remains an open problem: Its use has primarily been limited to relatively small…

2024

Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models

NeurIPS 2024poster

Diffusion models (DMs) have achieved remarkable success in text-to-image generation, but they also pose safety risks, such as the potential generation of harmful content and copyright violations. The techniques of machine unlearning, also known as concept erasing, have been developed to address thes…

2024

Revisiting Zeroth-Order Optimization for Memory-Efficient LLM Fine-Tuning: A Benchmark

ICML 2024poster

In the evolving landscape of natural language processing (NLP), fine-tuning pre-trained Large Language Models (LLMs) with first-order (FO) optimizers like SGD and Adam has become standard. Yet, as LLMs grow in size, the substantial memory overhead from back-propagation (BP) for FO gradient computati…

2024

SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning

EMNLP 2024main

Large Language Models (LLMs) have highlighted the necessity of effective unlearning mechanisms to comply with data regulations and ethical AI practices. LLM unlearning aims at removing undesired data influences and associated model capabilities without compromising utility beyond the scope of unlear…

2024

The Power of Few: Accelerating and Enhancing Data Reweighting with Coreset Selection

ICASSP 2024accepted

As machine learning tasks continue to evolve, the trend has been to gather larger datasets and train increasingly larger models. While this has led to advancements in accuracy, it has also escalated computational costs to unsustainable levels. Addressing this, our work aims to strike a delicate bala…

Cited by 0SourceScholar
2024

UnlearnCanvas: Stylized Image Dataset for Enhanced Machine Unlearning Evaluation in Diffusion Models

NeurIPS 2024poster

The technological advancements in diffusion models (DMs) have demonstrated unprecedented capabilities in text-to-image generation and are widely used in diverse applications. However, they have also raised significant societal concerns, such as the generation of harmful content and copyright dispute…

2023

RGMIL: Guide Your Multiple-Instance Learning Model with Regressor

NeurIPS 2023poster

In video analysis, an important challenge is insufficient annotated data due to the rare occurrence of the critical patterns, and we need to provide discriminative frame-level representation with limited annotation in some applications. Multiple Instance Learning (MIL) is suitable for this scenario.…

2023

Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learning

NeurIPS 2023poster

Massive data is often considered essential for deep learning applications, but it also incurs significant computational and infrastructural costs. Therefore, dataset pruning (DP) has emerged as an effective way to improve data efficiency by identifying and removing redundant training samples without…

2023

Text-Visual Prompting for Efficient 2D Temporal Video Grounding

CVPR 2023poster

In this paper, we study the problem of temporal video grounding (TVG), which aims to predict the starting/ending time points of moments described by a text sentence within a long untrimmed video. Benefiting from fine-grained 3D visual features, the TVG techniques have achieved remarkable progress in…

2022

How to Robustify Black-Box ML Models? A Zeroth-Order Optimization Perspective

ICLR 2022spotlight

The lack of adversarial robustness has been recognized as an important issue for state-of-the-art machine learning (ML) models, e.g., deep neural networks (DNNs). Thereby, robustifying ML models against adversarial attacks is now a major focus of research. However, nearly all existing defense method…

2022

Reverse Engineering of Imperceptible Adversarial Image Perturbations

ICLR 2022poster

It has been well recognized that neural network based image classifiers are easily fooled by images with tiny perturbations crafted by an adversary. There has been a vast volume of research to generate and defend such adversarial attacks. However, the following problem is left unexplored: How to rev…

2017

Transfer of View-manifold Learning to Similarity Perception of Novel Objects

ICLR 2017poster

We develop a model of perceptual similarity judgment based on re-training a deep convolution neural network (DCNN) that learns to associate different views of each 3D object to capture the notion of object persistence and continuity in our visual experience. The re-training process effectively perfo…

Cited by 11SourceScholar