← Search

Dong Li

127 accepted papers

2026

$\textbf{Re}^{2}$: Unlocking LLM Reasoning via Reinforcement Learning with Re-solving

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has shown promise in enhancing the reasoning performance of large language models (LLMs) by increasing test-time compute. However, even after extensive RLVR training, such models still tend to generate unnecessary and low-quality steps in their…

Cited by 0SourceScholar
2026

Boosting Multi-Domain Reasoning of LLMs via Curvature-Guided Policy Optimization

ICLR 2026poster

Multi-domain reinforcement learning (RL) for large language models (LLMs) involves highly intricate reward surfaces, posing significant challenges in finding parameters that excel across all domains. Recent empirical studies have further highlighted conflicts among domains, where gains in one capabi…

Cited by 0SourcecodeScholar
2026

CL-DMDF: Dynamic Multimodal Data Fusion Model Based on Contrastive Learning

AAAI 2026technical

Multimodal data fusion involves integrating and analyzing information from multiple modalities to uncover latent correlations and complementary patterns, thereby enhancing data processing and decision-making. While existing methods for structured multimodal inputs are typically designed around speci

Cited by 0SourcePDFScholar
2026

CompEvent: Complex-valued Event-RGB Fusion for Low-light Video Enhancement and Deblurring

AAAI 2026technical

Low-light video deblurring poses significant challenges in applications like nighttime surveillance and autonomous driving due to dim lighting and long exposures. While event cameras offer potential solutions with superior low-light sensitivity and high temporal resolution, existing fusion methods t

Cited by 0SourcePDFScholar
2026

DIFFSPARSE: ACCELERATING DIFFUSION TRANSFORMERS WITH LEARNED TOKEN SPARSITY

ICLR 2026poster

Diffusion models demonstrate outstanding performance in image generation, but their multi-step inference mechanism requires immense computational cost. Previous works accelerate inference by leveraging layer or token cache techniques to reduce computational cost. However, these methods fail to achie…

Cited by 0SourceScholar
2026

DiffBench Meets DiffAgent: End-to-End LLM-Driven Diffusion Acceleration Code Generation

AAAI 2026technical

Diffusion models have achieved remarkable success in image and video generation. However, their inherently multiple step inference process imposes substantial computational overhead, hindering real-world deployment. Accelerating diffusion models is therefore essential, yet determining how to combine

Cited by 0SourcePDFScholar
2026

Efficient Diffusion LLMs via Temporal-Spatial Parallel Decoding and Confidence Extrapolation

ICML 2026poster

Diffusion-based large language models (dLLMs) support parallel text generation via iterative denoising, yet inference remains latency-heavy because many steps are spent on redundant refinement and repeated remasking of tokens whose final values are already determined. Prior acceleration methods main…

Cited by 0SourceScholar
2026

Ego-InBetween: Generating Object State Transitions in Ego-Centric Videos

CVPR 2026

Understanding physical transformation processes is crucial for both human cognition and artificial intelligence systems, particularly from an egocentric perspective, which serves as a key bridge between humans and machines in action modeling. We define this modeling process as Egocentric Instructed

Cited by 0SourceScholar
2026

FinPercep-RM: A Fine-grained Reward Model and Co-evolutionary Curriculum for RL-based Real-world Super-Resolution

CVPR 2026

Inspired by the success of Reinforcement Learning with Human Feedback (RLHF) in image generation, recent work has adapted reward-based learning to image super-resolution (ISR) by using Image Quality Assessment (IQA) models as rewards. However, existing IQA models typically output only a single globa

Cited by 0SourcecodeScholar
2026

FreeInpaint: Tuning-free Prompt Alignment and Visual Rationality Enhancement in Image Inpainting

AAAI 2026technical

Text-guided image inpainting endeavors to generate new content within specified regions of images using textual prompts from users. The primary challenge is to accurately align the inpainted areas with the user-provided prompts while maintaining a high degree of visual fidelity. While existing inpai

Cited by 0SourcePDFScholar
2026

GauSem-SLAM: Gaussian Semantic Submaps with Loop Closure for Globally Consistent SLAM

ICRA 2026poster

3DGS has shown outstanding performance in multi-view geometry, driving its adoption in visual SLAM. However, real-time semantic 3DGS mapping faces challenges. Current methods typically treat semantics as external priors, making it hard to integrate them into SLAM tracking or loop closure correction.…

Cited by 0Scholar
2026

GenPRM: Scaling Test-Time Compute of Process Reward Models via Generative Reasoning

AAAI 2026technical

Recent advancements in Large Language Models (LLMs) have shown that it is promising to utilize Process Reward Models (PRMs) as verifiers to enhance the performance of LLMs. However, current PRMs face three key challenges: (1) limited process supervision and generalization capabilities, (2) dependenc

Cited by 0SourcePDFScholar
2026

LLM-Enhanced Energy Contrastive Learning for Out-of-Distribution Detection in Text-Attributed Graphs

AAAI 2026technical

Text-attributed graphs, where nodes are enriched with textual attributes, have become a powerful tool for modeling real-world networks such as citation, social, and transaction networks. However, existing methods for learning from these graphs often assume that the distributions of training and test

Cited by 0SourcePDFScholar
2026

Learnable Permutation for Structured Sparsity on Transformer Models

AAAI 2026technical

Structured sparsity has emerged as a popular model pruning technique, widely adopted in various architectures, including CNNs, Transformer models, and especially large language models (LLMs) in recent years. A promising direction to further improve post-pruning performance is weight permutation, whi

Cited by 0SourcePDFScholar
2026

MARLIN: Multi-Agent Reinforcement Learning for Incremental DAG Discovery

AAAI 2026technical

Uncovering causal structures from observational data is crucial for understanding complex systems and making informed decisions. While reinforcement learning (RL) has shown promise in identifying these structures in the form of a directed acyclic graph (DAG), existing methods often lack efficiency,

Cited by 0SourcePDFScholar
2026

Out-of-Distribution Detection with Positive and Negative Prompt Supervision Using Large Language Models

AAAI 2026technical

Out-of-distribution (OOD) detection is committed to delineating the classification boundaries between in-distribution (ID) and OOD images. Recent advances in vision-language models (VLMs) have demonstrated remarkable OOD detection performance by integrating both visual and textual modalities. In thi

Cited by 0SourcePDFScholar
2026

PARD: Accelerating LLM Inference with Low‑Cost PARallel Draft Model Adaptation

ICLR 2026poster

The autoregressive nature of large language models (LLMs) fundamentally limits inference speed, as each forward pass generates only a single token and is often bottlenecked by memory bandwidth. Speculative decoding has emerged as a promising solution, adopting a draft-then-verify strategy to acceler…

Cited by 0SourcecodeScholar
2026

PRISM: Synergizing Vision Foundation Models via Self-organized Expert Specialization

ICML 2026poster

Unifying the complementary strengths of diverse Vision Foundation Models (VFMs) into a single efficient model is highly desirable but challenged by the negative transfer inherent in monolithic distillation. To address these feature conflicts, we introduce \textbf{PRISM}, a novel dual-stream Mixture-…

Cited by 0SourceScholar
2026

PURE: Purging Unrelated Representations for Content-Agnostic Forgery Detection

IJCAI 2026

Existing AI-generated image (AIGI) detectors perform well in-domain but degrade severely under distribution shift. We observe that this failure is mainly caused by content shortcuts, where detectors spuriously couple forgery artifacts with semantic content, such as object categories or demographic a

Cited by 0Scholar
2026

SparK: Query-Aware Unstructured Sparsity with Recoverable KV Cache Channel Pruning

AAAI 2026technical

Long-context inference in large language models (LLMs) is increasingly constrained by the KV cache bottleneck: memory usage grows linearly with sequence length, while attention computation scales quadratically. Existing approaches address this issue by compressing the KV cache along the temporal axi

Cited by 0SourcePDFScholar
2026

SpikeVLA: Vision-Language-Action Models with Spiking Neural Networks

ICML 2026poster

Vision-Language-Action (VLA) models have become a central paradigm for embodied intelligence. However, most existing approaches are built on large-scale Transformers, resulting in substantial inference latency and energy consumption that limit their practical deployment in low-power, real-time scena…

Cited by 0SourceScholar
2026

Temporal-Synergistic Policy Optimization for Unsupervised Low-Light Image Enhancement

IJCAI 2026

Diffusion models show significant potential for low-light image enhancement. However, this task requires satisfying human perceptual preferences and content fidelity transcending simple brightness and color improvement. Existing methods rely on heuristic physical priors or incorporate perceptual met

Cited by 0Scholar
2026

Training-Free Loosely Speculative Decoding: Accepting Semantically Correct Drafts Beyond Exact Match

ICLR 2026poster

Large language models (LLMs) achieve strong performance across diverse tasks but suffer from high inference latency due to their autoregressive generation. Speculative Decoding (SPD) mitigates this issue by verifying candidate tokens from a smaller draft model in parallel, yet its strict exact-matc…

Cited by 0SourcecodeScholar
2025

A Lottery Ticket Hypothesis Approach with Sparse Fine-tuning and MAE for Image Forgery Detection and Localization

AAAI 2025technical

The rise in sophisticated image forgery techniques, driven by advancements in image editing and generation, has posed new security challenges. Traditional methods, designed for specific tampering artifacts, struggle with out-of-distribution image forgery detection. In this paper, we propose a shift…

2025

Amphista: Bi-directional Multi-head Decoding for Accelerating LLM Inference

NAACL 2025long

Large Language Models (LLMs) inherently use autoregressive decoding, which lacks parallelism in inference and results in significantly slow inference speed. While methods such as Medusa constructs parallelized heads, they lack adequate information interaction across different prediction positions. T…

Cited by 0SourcePDFScholar
2025

BaWA: Automatic Optimizing Pruning Metric for Large Language Models with Balanced Weight and Activation

ICML 2025poster

One-shot post-training pruning enhances the deployment of billion-scale large language models (LLMs), with the pruning metric playing a pivotal role in determining which weights to remove. However, existing metrics underperform due to their reliance on a simple symbolic combination of weights and ac…

Cited by 0SourcePDFScholar
2025

Bohdi: Heterogeneous LLM Fusion with Automatic Data Exploration

NeurIPS 2025poster

Heterogeneous Large Language Model (LLM) fusion integrates the strengths of multiple source LLMs with different architectures into a target LLM with low computational overhead. While promising, existing methods suffer from two major limitations: 1) **reliance on real data from limited domain** for k…

Cited by 0SourceScholar
2025

CRAFT: Time Series Forecasting with Cross-Future Behavior Awareness

IJCAI 2025

The past decades witness the significant advancements in time series forecasting (TSF) across various real-world domains, including e-commerce and disease spread prediction. However, TSF is usually constrained by the uncertainty dilemma of predicting future data with limited past observations. To se

2025

CoE: Chain-of-Explanation via Automatic Visual Concept Circuit Description and Polysemanticity Quantification

CVPR 2025poster

Explainability is a critical factor influencing the wide deployment of deep vision models (DVMs). Concept-based post-hoc explanation methods can provide both global and local insights into model decisions. However, current methods in this field face challenges in that they are inflexible to automati…

2025

Decouple to Reconstruct: High Quality UHD Restoration via Active Feature Disentanglement and Reversible Fusion

ICCV 2025poster

Ultra-high-definition (UHD) image restoration often faces computational bottlenecks and information loss due to its extremely high resolution. Existing studies based on Variational Autoencoders (VAE) improve efficiency by transferring the image restoration process from pixel space to latent space. H…

Cited by 0SourcePDFScholar
2025

DiffZOO: A Purely Query-Based Black-Box Attack for Red-teaming Text-to-Image Generative Model via Zeroth Order Optimization

NAACL 2025findings

Current text-to-image (T2I) synthesis diffusion models raise misuse concerns, particularly in creating prohibited or not-safe-for-work (NSFW) images. To address this, various safety mechanisms and red teaming attack methods are proposed to enhance or expose the T2I model’s capability to generate uns…

2025

DreamUHD: Frequency Enhanced Variational Autoencoder for Ultra-High-Definition Image Restoration

AAAI 2025technical

Existing ultra-high-definition (UHD) image restoration methods often struggle with consistency due to downsampling. We aim to address these challenges by leveraging the powerful latent space representation and reconstruction capabilities of Variational Autoencoders (VAE). However, applying VAE to UH…

2025

EGSRAL:An Enhanced 3D Gaussian Splatting Based Renderer with Automated Labeling for Large-Scale Driving Scene

AAAI 2025technical

3D Gaussian Splatting (3D GS) has gained popularity due to its faster rendering speed and high-quality novel view synthesis. Some researchers have explored using 3D GS for reconstructing driving scenes. However, these methods often rely on various types of data, such as depth maps, 3D bounding boxes…

2025

Enhanced Pansharpening via Quaternion Spatial-Spectral Interactions

ICCV 2025poster

Pansharpening aims to generate high-resolution multispectral (MS) images by fusing panchromatic (PAN) images with corresponding low-resolution MS images. However, many existing methods struggle to fully capture spatial and spectral interactions, limiting their effectiveness. To address this, we prop…

2025

Enhancing One-Shot Pruned Pre-trained Language Models through Sparse-Dense-Sparse Mechanism

COLING 2025main

Pre-trained language models (PLMs) are engineered to be robust in contextual understanding and exhibit outstanding performance in various natural language processing tasks. However, their considerable size incurs significant computational and storage costs. Modern pruning strategies employ retrainin…

Cited by 0SourcePDFScholar
2025

FADE: Towards Fairness-aware Data Generation for Domain Generalization via Classifier-Guided Score-based Diffusion Models

IJCAI 2025

Fairness-aware domain generalization (FairDG) has emerged as a critical challenge for deploying trustworthy AI systems, particularly in scenarios involving distribution shifts. Traditional methods for addressing fairness have failed in domain generalization due to their lack of consideration for dis

Cited by 0SourcePDFScholar
2025

Fast and Slow Gradient Approximation for Binary Neural Network Optimization

AAAI 2025technical

Binary Neural Networks (BNNs) have garnered significant attention due to their immense potential for deployment on edge devices. However, the non-differentiability of the quantization function poses a challenge for the optimization of BNNs, as its derivative cannot be backpropagated. To address this…

2025

FourierMamba: Fourier Learning Integration with State Space Models for Image Deraining

ICML 2025poster

Image deraining aims to remove rain streaks from rainy images and restore clear backgrounds. Currently, some research that employs the Fourier transform has proved to be effective for image deraining, due to it acting as an effective frequency prior for capturing rain streaks. However, despite there…

Cited by 17SourcePDFScholar
2025

FreeDNA: Endowing Domain Adaptation of Diffusion-Based Dense Prediction with Training-Free Domain Noise Alignment

ICCV 2025poster

Domain Adaptation (DA) for dense prediction tasks is an important topic, which enhances the dense prediction model's performance when tested on its unseen domain. Recently, with the development of Diffusion-based Dense Prediction (DDP) models, the exploration of DA designs tailored to this framework…

2025

GDDA: Semantic OOD Detection on Graphs under Covariate Shift via Score-Based Diffusion Models

ICASSP 2025accepted

Out-of-distribution (OOD) detection poses a signifi-cant challenge for Graph Neural Networks (GNNs), particularly in open-world scenarios with varying distribution shifts. Most existing OOD detection methods on graphs primarily focus on identifying instances in test data domains caused by either sem…

Cited by 0SourceScholar
2025

GaussR-SLAM: Gaussian Robust SLAM in Data Loss and Interference Environments

RA-L 2025

Recent advancements in 3DGS-based explicit mapping have significantly improved SLAM performance, achieving more realistic environment reconstruction and faster processing. However, issues such as data loss caused by unstable data transmission, textureless and repetitive-texture often occur in real-w

Cited by 0SourceScholar
2025

Graph Counselor: Adaptive Graph Exploration via Multi-Agent Synergy to Enhance LLM Reasoning

ACL 2025long

Graph Retrieval Augmented Generation (GraphRAG) effectively enhances external knowledge integration capabilities by explicitly modeling knowledge relationships, thereby improving the factual accuracy and generation quality of Large Language Models (LLMs) in specialized domains. However, existing met…

2025

Gumiho: A Hybrid Architecture to Prioritize Early Tokens in Speculative Decoding

ICML 2025poster

Speculative decoding (SPD) aims to accelerate the auto-regressive token generation process of a target Large Language Model (LLM). Some approaches employ a draft model with multiple heads to predict a sequence of future tokens, where each head handles a token in the sequence. The target LLM verifies…

2025

KELE: A Multi-Agent Framework for Structured Socratic Teaching with Large Language Models

EMNLP 2025

Socratic teaching, known for its emphasis on heuristic questioning and deep thinking, has demonstrated significant advantages in promoting students’ cognitive development. However, traditional Socratic teaching places high demands on teachers’ expertise and real-time feedback capabilities, making it

2025

Latent Harmony: Synergistic Unified UHD Image Restoration via Latent Space Regularization and Controllable Refinement

NeurIPS 2025poster

Ultra-High Definition (UHD) image restoration struggles to balance computational efficiency and detail retention. While Variational Autoencoders (VAEs) offer improved efficiency by operating in the latent space, with the Gaussian variational constraint, this compression preserves semantics but sacri…

Cited by 0SourceScholar
2025

Learnable Frequency Decomposition for Image Forgery Detection and Localization

IJCAI 2025

Concern for image authenticity spurs research in image forgery detection and localization (IFDL). Most deep learning-based methods focus primarily on spatial domain modeling and have not fully explored frequency domain strategies. In this paper, we observe and analyze the frequency characteristic ch

Cited by 0SourcePDFScholar
2025

Less is More: Efficient Model Merging with Binary Task Switch

CVPR 2025highlight

As an effective approach to equip models with multi-task capabilities without additional training, model merging has garnered significant attention. However, existing merging methods face challenges of redundant parameter conflicts and the excessive storage burden of fine-tuned parameters. In this w…

Cited by 1SourcePDFScholar
2025

MigGPT: Harnessing Large Language Models for Automated Migration of Out-of-Tree Linux Kernel Patches Across Versions

NeurIPS 2025spotlight

Out-of-tree kernel patches are essential for adapting the Linux kernel to new hardware or enabling specific functionalities. Maintaining and updating these patches across different kernel versions demands significant effort from experienced engineers. Large language models (LLMs) have shown remarkab…

Cited by 0SourceScholar
2025

PanComplex: Leveraging Complex-Valued Neural Networks for Enhanced Pansharpening

IJCAI 2025

Pansharpening combines panchromatic and low-resolution multispectral images to generate high-resolution multispectral images. Previous studies have explored the connection between pansharpening and the frequency domain, but mostly in the real-valued domain, leaving the complex domain relatively unex

2025

Pursuing Temporal-Consistent Video Virtual Try-On via Dynamic Pose Interaction

CVPR 2025poster

Video virtual try-on aims to seamlessly dress a subject in a video with a specific garment. The primary challenge involves preserving the visual authenticity of the garment while dynamically adapting to the pose and physique of the subject. While existing methods have predominantly focused on image-…

Cited by 0SourcePDFScholar
2025

ReNeg: Learning Negative Embedding with Reward Guidance

CVPR 2025highlight

In text-to-image (T2I) generation applications, negative embeddings have proven to be a simple yet effective approach for enhancing generation quality. Typically, these negative embeddings are derived from user-defined negative prompts, which, while being functional, are not necessarily optimal. In…

2025

Reinforced In-Context Black-Box Optimization

IJCAI 2025

Black-Box Optimization (BBO) has found successful applications in many fields of science and engineering. Recently, there has been a growing interest in meta-learning particular components of BBO algorithms to speed up optimization and get rid of tedious hand-crafted heuristics. As an extension, lea

2025

STG-Avatar: Animatable Human Avatars via Spacetime Gaussian

IROS 2025

Realistic animatable human avatars from monocular videos are crucial for advancing human-robot interaction and enhancing immersive virtual experiences. While recent research on 3DGS-based human avatars has made progress, it still struggles with accurately representing detailed features of non-rigid

Cited by 6SourcecodeScholar
2025

SolverLLM: Leveraging Test-Time Scaling for Optimization Problem via LLM-Guided Search

NeurIPS 2025poster

Large Language Models (LLMs) offer promising capabilities for tackling complex reasoning tasks, including optimization problems. However, existing methods either rely on prompt engineering, which leads to poor generalization across problem types, or require costly supervised training. We introduce S…

Cited by 0SourceScholar
2025

TWC-SLAM: Multi-Agent Cooperative SLAM with Text Semantics and WiFi Features Integration for Similar Indoor Environments

IROS 2025

Multi-agent cooperative SLAM often encounters challenges in similar indoor environments characterized by repetitive structures, such as corridors and rooms. These challenges can lead to significant inaccuracies in shared location identification when employing point cloud-based techniques. To mitigat

Cited by 0SourceScholar
2025

Týr-the-Pruner: Structural Pruning LLMs via Global Sparsity Distribution Optimization

NeurIPS 2025poster

Structural pruning enhances hardware-agnostic inference efficiency for large language models (LLMs) yet often fails to maintain comparable performance. Local pruning performs efficient layer-by-layer compression but ignores global topology. Although global pruning aims to identify an optimal sparse…

Cited by 0SourceScholar
2025

UHD-processer: Unified UHD Image Restoration with Progressive Frequency Learning and Degradation-aware Prompts

CVPR 2025poster

We introduce UHD-Processor, a unified and robust framework for all-in-one image restoration, which is particularly resource-efficient for Ultra-High-Definition (UHD) images. To address the limitations of traditional all-in-one methods that rely on complex restoration backbones, our strategy employs…

2024

An Efficient Memory Module for Graph Few-Shot Class-Incremental Learning

NeurIPS 2024poster

Graph incremental learning has gained widespread attention for its ability to mitigate catastrophic forgetting for graph neural networks (GNN). Conventional methods typically require numerous labels for node classification. However, obtaining abundant labels is often challenging in practice, which m…

2024

CO2: Efficient Distributed Training with Full Communication-Computation Overlap

ICLR 2024spotlight

The fundamental success of large language models hinges upon the efficacious implementation of large-scale distributed training techniques. Nevertheless, building a vast, high-performance cluster featuring high-speed communication interconnectivity is prohibitively costly, and accessible only to pro…

2024

CatmullRom Splines-Based Regression for Image Forgery Localization

AAAI 2024technical

IFL (Image Forgery Location) helps secure digital media forensics. However, many methods suffer from false detections (i.e., FPs) and inaccurate boundaries. In this paper, we proposed the CatmullRom Splines-based Regression Network (CSR-Net), which first rethinks the IFL task from the perspective of…

Cited by 14SourcePDFScholar
2024

DL-QAT: Weight-Decomposed Low-Rank Quantization-Aware Training for Large Language Models

EMNLP 2024industry

Improving the efficiency of inference in Large Language Models (LLMs) is a critical area of research. Post-training Quantization (PTQ) is a popular technique, but it often faces challenges at low-bit levels, particularly in downstream tasks. Quantization-aware Training (QAT) can alleviate this probl…

Cited by 2SourcePDFScholar
2024

DR-Label: Label Deconstruction and Reconstruction of GNN Models for Catalysis Systems

AAAI 2024technical

Attaining the equilibrium geometry of a catalyst-adsorbate system is key to fundamentally assessing its effective properties, such as adsorption energy. While machine learning methods with advanced representation or supervision strategies have been applied to boost and guide the relaxation processes…

2024

DiP-GO: A Diffusion Pruner via Few-step Gradient Optimization

NeurIPS 2024poster

Diffusion models have achieved remarkable progress in the field of image generation due to their outstanding capabilities. However, these models require substantial computing resources because of the multi-step denoising process during inference. While traditional pruning methods have been employed…

Cited by 2SourcePDFScholar
2024

Enhancing Vision Transformer: Amplifying Non-Linearity in Feedforward Network Module

ICML 2024poster

Transformer models have been gaining substantial interest in the field of computer vision tasks nowadays. Although a vision transformer contains two important components which are self-attention module and feedforward network (FFN) module, the majority of research tends to concentrate on modifying t…

Cited by 3SourcePDFScholar
2024

FinBen: A Holistic Financial Benchmark for Large Language Models

NeurIPS 2024poster

LLMs have transformed NLP and shown promise in various fields, yet their potential in finance is underexplored due to a lack of comprehensive benchmarks, the rapid development of LLMs, and the complexity of financial tasks. In this paper, we introduce FinBen, the first extensive open-source evaluati…

2024

FinCon: A Synthesized LLM Multi-Agent System with Conceptual Verbal Reinforcement for Enhanced Financial Decision Making

NeurIPS 2024poster

Large language models (LLMs) have demonstrated notable potential in conducting complex tasks and are increasingly utilized in various financial applications. However, high-quality sequential financial investment decision-making remains challenging. These tasks require multiple interactions with a vo…

2024

HarmonyDream: Task Harmonization Inside World Models

ICML 2024poster

Model-based reinforcement learning (MBRL) holds the promise of sample-efficient learning by utilizing a world model, which models how the environment works and typically encompasses components for two tasks: observation modeling and reward modeling. In this paper, through a dedicated empirical inves…

2024

HomoFormer: Homogenized Transformer for Image Shadow Removal

CVPR 2024poster

The spatial non-uniformity and diverse patterns of shadow degradation conflict with the weight sharing manner of dominant models which may lead to an unsatisfactory compromise. To tackle with this issue we present a novel strategy from the view of shadow transformation in this paper: directly homoge…

2024

Improving Unsupervised Hierarchical Representation with Reinforcement Learning

CVPR 2024poster

Learning representations to capture the very fundamental understanding of the world is a key challenge in machine learning. The hierarchical structure of explanatory factors hidden in data is such a general representation and could be potentially achieved with a hierarchical VAE. However training a…

2024

Interactive Continual Learning: Fast and Slow Thinking

CVPR 2024poster

Advanced life forms sustained by the synergistic interaction of neural cognitive mechanisms continually acquire and transfer knowledge throughout their lifespan. In contrast contemporary machine learning paradigms exhibit limitations in emulating the facets of continual learning (CL). Nonetheless th…

2024

Learning Discriminative Noise Guidance for Image Forgery Detection and Localization

AAAI 2024technical

This study introduces a new method for detecting and localizing image forgery by focusing on manipulation traces within the noise domain. We posit that nearly invisible noise in RGB images carries tampering traces, useful for distinguishing and locating forgeries. However, the advancement of tamperi…

Cited by 16SourcePDFScholar
2024

Multiagent Gumbel MuZero: Efficient Planning in Combinatorial Action Spaces

AAAI 2024technical

AlphaZero and MuZero have achieved state-of-the-art (SOTA) performance in a wide range of domains, including board games and robotics, with discrete and continuous action spaces. However, to obtain an improved policy, they often require an excessively large number of simulations, especially for doma…

2024

Noise-assisted Prompt Learning for Image Forgery Detection and Localization

ECCV 2024poster

"We present CLIP-IFDL, a novel image forgery detection and localization (IFDL) model that harnesses the power of Contrastive Language Image Pre-Training (CLIP). However, directly incorporating CLIP in forgery detection poses challenges, given its lack of specific prompts and forgery consciousness. T…

Cited by 4SourcePDFScholar
2024

PG-LBO: Enhancing High-Dimensional Bayesian Optimization with Pseudo-Label and Gaussian Process Guidance

AAAI 2024technical

Variational Autoencoder based Bayesian Optimization (VAE-BO) has demonstrated its excellent performance in addressing high-dimensional structured optimization problems. However, current mainstream methods overlook the potential of utilizing a pool of unlabeled data to construct the latent space, whi…

2024

QT-ViT: Improving Linear Attention in ViT with Quadratic Taylor Expansion

NeurIPS 2024poster

Vision transformer model (ViT) is widely used and performs well in vision tasks due to its ability to capture long-range dependencies. However, the time complexity and memory consumption increase quadratically with the number of input patches which limits the usage of ViT in real-world applications.…

Cited by 0SourcePDFScholar
2024

SMR: State Memory Replay for Long Sequence Modeling

ACL 2024findings

Despite the promising performance of state space models (SSMs) in long sequence modeling, limitations still exist. Advanced SSMs like S5 and S6 (Mamba) in addressing non-uniform sampling, their recursive structures impede efficient SSM computation via convolution. To overcome compatibility limitatio…

2024

Sample-Efficient Quality-Diversity by Cooperative Coevolution

ICLR 2024spotlight

Quality-Diversity (QD) algorithms, as a subset of evolutionary algorithms, have emerged as a powerful optimization paradigm with the aim of generating a set of high-quality and diverse solutions. Although QD has demonstrated competitive performance in reinforcement learning, its low sample efficienc…

Cited by 11SourcePDFScholar
2024

Scaling Laws for Linear Complexity Language Models

EMNLP 2024main

The interest in linear complexity models for large language models is on the rise, although their scaling capacity remains uncertain. In this study, we present the scaling laws for linear complexity language models to establish a foundation for their scalability. Specifically, we examine the scaling…

2024

Supervised Algorithmic Fairness in Distribution Shifts: A Survey

IJCAI 2024poster

Supervised fairness-aware machine learning under distribution shifts is an emerging field that addresses the challenge of maintaining equitable and unbiased predictions when faced with changes in data distributions from source to target domains. In real-world applications, machine learning models a…

Cited by 12SourcePDFScholar
2024

Towards Generalizable Neural Solvers for Vehicle Routing Problems via Ensemble with Transferrable Local Policy

IJCAI 2024poster

Machine learning has been adapted to help solve NP-hard combinatorial optimization problems. One prevalent way is learning to construct solutions by deep neural networks, which has been receiving more and more attention due to the high efficiency and less requirement for expert knowledge. However, m…

2024

UPDP: A Unified Progressive Depth Pruner for CNN and Vision Transformer

AAAI 2024technical

Traditional channel-wise pruning methods by reducing network channels struggle to effectively prune efficient CNN models with depth-wise convolutional layers and certain efficient modules, such as popular inverted residual blocks. Prior depth pruning methods by reducing network depths are not suitab…

Cited by 13SourcePDFScholar
2024

Various Lengths, Constant Speed: Efficient Language Modeling with Lightning Attention

ICML 2024poster

We present Lightning Attention, the first linear attention implementation that maintains a constant training speed for various sequence lengths under fixed memory consumption. Due to the issue with cumulative summation operations (cumsum), previous linear attention implementations cannot achieve the…

2024

iVideoGPT: Interactive VideoGPTs are Scalable World Models

NeurIPS 2024poster

World models empower model-based agents to interactively explore, reason, and plan within imagined environments for real-world decision-making. However, the high demand for interactivity poses challenges in harnessing recent advancements in video generative models for developing world models at scal…

2023

6D Pose Estimation Based on 3D Edge Binocular Reprojection Optimization for Robotic Assembly

RA-L 2023

Accurate 6D pose estimation of object is important for robot assembly. This letter presents a novel method for achieving high precision 6D pose estimation by exploiting the reprojection of 3D edges onto binocular RGB image pairs. Our proposed method encompasses three phases: detection, pose initiali

Cited by 9SourceScholar
2023

Boosting Multiagent Reinforcement Learning via Permutation Invariant and Permutation Equivariant Networks

ICLR 2023poster

The state space in Multiagent Reinforcement Learning (MARL) grows exponentially with the agent number. Such a curse of dimensionality results in poor scalability and low sample efficiency, inhibiting MARL for decades. To break this curse, we propose a unified agent permutation framework that exploit…

Cited by 30SourcePDFScholar
2023

Clustering-Based Supervised Contrastive Learning for Identifying Risk Items on Heterogeneous Graph

ICASSP 2023accepted

Risk item identification is vital for protecting the health of ecommerce trades. Existing solutions prefer to model structure information besides item attributes and optimize parameters in cross-entropy (CE) manners. However, the few labeled and imbalanced supervision in real-world scenarios usually…

Cited by 0SourceScholar
2023

Cooperative Control Strategy of Magnetic Microrobots in Bifurcated Microfluidic Channels

RA-L 2023

Magnetically actuated nanoparticle microrobots have great potential for biomedical applications, especially in blood vessels. However, they face a complex bifurcation environment, and magnetic control strategies targeting a single pathway limit the operational efficiency of the targeting task. In th

Cited by 5SourceScholar
2023

FDViT: Improve the Hierarchical Architecture of Vision Transformer

ICCV 2023poster

Despite the fact that transformer-based models have yielded great success in computer vision tasks, they suffer from the challenge of high computational costs that limits their use on resource-constrained devices. One major reason is that vision transformers have redundant calculations since the sel…

Cited by 18PDFScholar
2023

Fine-Grained Audible Video Description

CVPR 2023poster

We explore a new task for audio-visual-language modeling called fine-grained audible video description (FAVD). It aims to provide detailed textual descriptions for the given audible videos, including the appearance and spatial locations of each object, the actions of moving objects, and the sounds i…

2023

Perturbation Towards Easy Samples Improves Targeted Adversarial Transferability

NeurIPS 2023poster

The transferability of adversarial perturbations provides an effective shortcut for black-box attacks. Targeted perturbations have greater practicality but are more difficult to transfer between models. In this paper, we experimentally and theoretically demonstrated that neural networks trained on t…

2023

Reining Generalization in Offline Reinforcement Learning via Representation Distinction

NeurIPS 2023poster

Offline Reinforcement Learning (RL) aims to address the challenge of distribution shift between the dataset and the learned policy, where the value of out-of-distribution (OOD) data may be erroneously estimated due to overgeneralization. It has been observed that a considerable portion of the benefi…

Cited by 7SourcePDFScholar
2023

Toeplitz Neural Network for Sequence Modeling

ICLR 2023top-25%

Sequence modeling has important applications in natural language processing and computer vision. Recently, the transformer-based models have shown strong performance on various sequence modeling tasks, which rely on attention to capture pairwise token relations, and position embedding to inject posi…

2023

Towards Reliable Item Sampling for Recommendation Evaluation

AAAI 2023technical

Since Rendle and Krichene argued that commonly used sampling-based evaluation metrics are ``inconsistent'' with respect to the global metrics (even in expectation), there have been a few studies on the sampling-based recommender system evaluation. Existing methods try either mapping the sampling-bas…

Cited by 12SourcePDFScholar
2022

Cross-Dataset Collaborative Learning for Semantic Segmentation in Autonomous Driving

AAAI 2022technical

Semantic segmentation is an important task for scene understanding in self-driving cars and robotics, which aims to assign dense labels for all pixels in the image. Existing work typically improves semantic segmentation performance by exploring different network architectures on a target dataset. Li…

Cited by 44SourcePDFScholar
2022

Dual Cross-Attention Learning for Fine-Grained Visual Categorization and Object Re-Identification

CVPR 2022poster

Recently, self-attention mechanisms have shown impressive performance in various NLP and CV tasks, which can help capture sequential characteristics and derive global information. In this work, we explore how to extend self-attention modules to better learn subtle feature embeddings for recognizing…

Cited by 230PDFScholar
2022

Learning Pseudometric-based Action Representations for Offline Reinforcement Learning

ICML 2022spotlight

Offline reinforcement learning is a promising approach for practical applications since it does not require interactions with real-world environments. However, existing offline RL methods only work well in environments with continuous or small discrete action spaces. In environments with large and d…

Cited by 20SourcePDFScholar
2022

Learning State Representations via Retracing in Reinforcement Learning

ICLR 2022poster

We propose learning via retracing, a novel self-supervised approach for learning the state representation (and the associated dynamics model) for reinforcement learning tasks. In addition to the predictive (reconstruction) supervision in the forward direction, we propose to include "retraced" transi…

2022

Multi-Document Scientific Summarization from a Knowledge Graph-Centric View

COLING 2022main

Multi-Document Scientific Summarization (MDSS) aims to produce coherent and concise summaries for clusters of topic-relevant scientific papers. This task requires precise understanding of paper content and accurate modeling of cross-paper relationships. Knowledge graphs convey compact and interpreta…

2022

Neuro-Symbolic Hierarchical Rule Induction

ICML 2022spotlight

We propose Neuro-Symbolic Hierarchical Rule Induction, an efficient interpretable neuro-symbolic model, to solve Inductive Logic Programming (ILP) problems. In this model, which is built from a pre-defined set of meta-rules organized in a hierarchical structure, first-order rules are invented by lea…

2022

Versatile Multi-stage Graph Neural Network for Circuit Representation

NeurIPS 2022accept

Due to the rapid growth in the scale of circuits and the desire for knowledge transfer from old designs to new ones, deep learning technologies have been widely exploited in Electronic Design Automation (EDA) to assist circuit design. In chip design cycles, we might encounter heterogeneous and diver…

Cited by 39SourcePDFScholar
2022

What about Inputting Policy in Value Function: Policy Representation and Policy-Extended Value Function Approximator

AAAI 2022technical

We study Policy-extended Value Function Approximator (PeVFA) in Reinforcement Learning (RL), which extends conventional value function approximator (VFA) to take as input not only the state (and action) but also an explicit policy representation. Such an extension enables PeVFA to preserve values of…

Cited by 26SourcePDFScholar
2021

Adaptive Online Packing-guided Search for POMDPs

NeurIPS 2021poster

The partially observable Markov decision process (POMDP) provides a general framework for modeling an agent's decision process with state uncertainty, and online planning plays a pivotal role in solving it. A belief is a distribution of states representing state uncertainty. Methods for large-scale…

2021

An Efficient Transfer Learning Framework for Multiagent Reinforcement Learning

NeurIPS 2021poster

Transfer Learning has shown great potential to enhance single-agent Reinforcement Learning (RL) efficiency. Similarly, Multiagent RL (MARL) can also be accelerated if agents can share knowledge with each other. However, it remains a problem of how an agent should learn from other agents. In this pap…

2021

On Estimating Recommendation Evaluation Metrics under Sampling

AAAI 2021technical

Since the recent studies (KDD'20) done by Krichene and Rendle on the sampling based top-k evaluation metric for recommendation, there have been a lot of debate on the validity of using sampling for evaluating recommendation algorithms. Though their work and the recent work done by Li et. al. (KDD'…

Cited by 16SourcePDFScholar
2021

ROBI: A Multi-View Dataset for Reflective Objects in Robotic Bin-Picking

IROS 2021poster

In robotic bin-picking applications, the perception of texture-less, highly reflective parts is a valuable but challenging task. The high glossiness can introduce fake edges in RGB images and inaccurate depth measurements, especially in heavily cluttered bin scenarios. In this paper, we present the…

Cited by 55SourceScholar
2021

Representing Videos As Discriminative Sub-Graphs for Action Recognition

CVPR 2021poster

Human actions are typically of combinatorial structures or patterns, i.e., subjects, objects, plus spatio-temporal interactions in between. Discovering such structures is therefore a rewarding way to reason about the dynamics of interactions and recognize the actions. In this paper, we introduce a n…

Cited by 34PDFScholar
2021

Towards Discriminative Representation Learning for Unsupervised Person Re-Identification

ICCV 2021poster

In this work, we address the problem of unsupervised domain adaptation for person re-ID where annotations are available for the source domain but not for target. Previous methods typically follow a two-stage optimization pipeline, where the network is first pre-trained on source and then fine-tuned…

Cited by 86PDFScholar
2021

Towards Effective Context for Meta-Reinforcement Learning: an Approach based on Contrastive Learning

AAAI 2021technical

Context, the embedding of previous collected trajectories, is a powerful construct for Meta-Reinforcement Learning (Meta-RL) algorithms. By conditioning on an effective context, Meta-RL policies can easily generalize to new tasks within a few adaptation steps. We argue that improving the quality of…

2020

HM-ANN: Efficient Billion-Point Nearest Neighbor Search on Heterogeneous Memory

NeurIPS 2020poster

The state-of-the-art approximate nearest neighbor search (ANNS) algorithms face a fundamental tradeoff between query latency and accuracy, because of small main memory capacity: To store indices in main memory for short query latency, the ANNS algorithms have to limit dataset size or use a quantizat…

2020

ProgressFace: Scale-Aware Progressive Learning for Face Detection

ECCV 2020poster

Scale variation stands out as one of key challenges in face detection. Recent attempts have been made to cope with this issue by incorporating image / feature pyramids or adjusting anchor sampling / matching strategies. In this work, we propose a novel scale-aware progressive training mechanism to a…

2018

Recurrent Tubelet Proposal and Recognition Networks for Action Detection

ECCV 2018poster

Detecting actions in videos is a challenging task as video is an information intensive media with complex variations. Existing approaches predominantly generate action proposals for each individual frame or fixed-length clip independently, while overlooking temporal context across them. Such tempora…

Cited by 146SourcePDFScholar
2016

Weakly Supervised Object Localization With Progressive Domain Adaptation

CVPR 2016poster

We address the problem of weakly supervised object localization where only image-level annotations are available for training. Many existing approaches tackle this problem through object proposal mining. However, a substantial amount of noise in object proposals causes ambiguities for learning discr…

Cited by 257PDFScholar