← Search

Dongsheng Li

153 accepted papers

2026

Do Not Let Low-Probability Tokens Over-Dominate in RL for LLMs

ICLR 2026poster

Reinforcement learning (RL) has become a cornerstone for enhancing the reasoning capabilities of large language models (LLMs), with recent innovations such as Group Relative Policy Optimization (GRPO) demonstrating exceptional effectiveness. In this study, we identify a critical yet underexplored is…

Cited by 0SourcecodeScholar
2026

EgoBrain: Synergizing Minds and Eyes For Human Action Understanding

ICLR 2026poster

The integration of brain-computer interfaces (BCIs), in particular electroencephalography (EEG), with artificial intelligence (AI) has shown tremendous promise in decoding human cognition and behavior from neural signals. In particular, the rise of multimodal AI models have brought new possibilities…

Cited by 0SourcecodeScholar
2026

Exploratory Memory-Augmented LLM Agent via Hybrid On- and Off-Policy Optimization

ICLR 2026poster

Exploration remains the key bottleneck for large language model agents trained with reinforcement learning. While prior methods exploit pretrained knowledge, they fail in environments requiring the discovery of novel states. We propose EMPO$^2$, a hybrid RL framework that leverages memory for explor…

Cited by 0SourcecodeScholar
2026

Exploring Interpretability for Visual Prompt Tuning with Cross-layer Concepts

ICLR 2026poster

Visual prompt tuning offers significant advantages for adapting pre-trained visual foundation models to specific tasks. However, current research provides limited insight into the interpretability of this approach, which is essential for enhancing AI reliability and enabling AI-driven knowledge disc…

Cited by 0SourcecodeScholar
2026

Improving Diffusion Models for Class-imbalanced Training Data via Capacity Manipulation

ICLR 2026oral

While diffusion models have achieved remarkable performance in image generation, they often struggle with the imbalanced datasets frequently encountered in real-world applications, resulting in significant performance degradation on minority classes. In this paper, we identify model capacity allocat…

Cited by 0SourceScholar
2026

Improving Diffusion Planners by Self-Supervised Action Gating with Energies

ICML 2026poster

Diffusion planners are a strong approach for offline reinforcement learning, but they can fail when value-guided selection favours trajectories that score well yet are locally inconsistent with the environment dynamics, resulting in brittle execution. We propose Self-supervised Action Gating with En…

Cited by 0SourceScholar
2026

Improving Long-Context Summarization with Multi-Granularity Retrieval Optimization

AAAI 2026technical

Retrieval-Augmented Generation (RAG) is an effective solution to overcome the limitations of Large Language Models (LLMs) in terms of specific-domain knowledge and timely information updates. However, current RAG methods typically respond to queries based on isolated segments, lacking the ability to

Cited by 0SourcePDFScholar
2026

Joint Adaptation of Uni-modal Foundation Models for Multi-modal Alzheimer's Disease Diagnosis

ICLR 2026poster

Alzheimer’s Disease (AD) is a progressive neurodegenerative disorder and a leading cause of dementia worldwide. Accurate diagnosis requires integrating diverse patient data modalities. With the rapid advancement of foundation models in neurobiology and medicine, integrating foundation models from va…

Cited by 0SourceScholar
2026

Kuramoto Oscillatory Phase Encoding: Neuro-inspired Synchronization for Improved Learning Efficiency

ICML 2026poster

Spatiotemporal neural dynamics and oscillatory synchronization are widely implicated in biological information processing and have been hypothesized to support flexible coordination such as feature binding. By contrast, most deep learning architectures represent and propagate information through act…

Cited by 0SourceScholar
2026

Let Your Image Move with Your Motion! -- Implicit Multi-Object Multi-Motion Transfer

CVPR 2026

Motion transfer has emerged as a promising direction for controllable video generation, yet existing methods largely focus on single-object scenarios and struggle when multiple objects require distinct motion patterns. In this work, we present FlexiMMT, the first implicit image-to-video (I2V) motion

Cited by 0SourcecodeScholar
2026

Mitigating Plasticity Loss through Architectural Design in Continual Learning

ICML 2026poster

Neural networks for continual reinforcement learning (CRL) often suffer from plasticity loss, i.e., a progressive decline in their ability to learn new tasks arising from increased representational drift (churn) and Neural Tangent Kernel (NTK) rank collapse. Current methods mitigating this problem i…

Cited by 0SourceScholar
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

Quota-Calibrated Fine-Grained Alignment with Context-Aware Marginals for Text-based Person Retrieval

CVPR 2026

The core challenge in Text-based Person Retrieval (TPR) lies in establishing fine-grained, many-to-many semantic alignment between textual words and visual regions. Existing methods predominantly rely on pointwise similarity or attention mechanisms, implicitly assuming matches are independent and ba

Cited by 0SourceScholar
2026

Reasoning-Driven Multimodal LLM for Domain Generalization

ICLR 2026poster

This paper addresses the domain generalization (DG) problem in deep learning. While most DG methods focus on enforcing visual feature invariance, we leverage the reasoning capability of multimodal large language models (MLLMs) and explore the potential of constructing reasoning chains that derives…

Cited by 0SourceScholar
2026

Revealing Modular Gradient Noise Imbalance in LLMs: Calibrating Adam via Signal-to-Noise Ratio

IJCAI 2026

The impressive performance of large language models (LLMs) arises from their massive scale and heterogeneous module composition. However, this structural heterogeneity poses significant optimization challenges. While adaptive optimizers such as Adam(W) provide per-parameter adaptivity, they do not e

Cited by 0Scholar
2026

Stabilized Supralinear Networks Learn to Switch Coding Strategies Balancing Cost and Performance

ICML 2026poster

Lateral connections (LCs) are ubiquitous in the cortical circuits. While modern deep learning architectures have rich intralayer interactions (e.g., convolutional mixing, normalization, or attention) to support feature selectivity and contextual modulation, explicit excitatory and inhibitory (E-I) L…

Cited by 0SourceScholar
2026

TiME: Test-Time Mixture-of-Experts Routing via Asymmetric CO-Optimal Transport for Continual Test-Time Adaptation

ICML 2026poster

Large language models usually face continuous domain shifts during testing, which degrade performance on unseen shifting domains. So, researchers propose continual test-time adaptation (CTTA) to adapt to evolving testing domains while preserving knowledge of previous domains, making adaptability-sta…

Cited by 0SourceScholar
2026

Trade in Minutes! Rationality-Driven Agentic System for Quantitative Financial Trading

ICLR 2026poster

Recent advancements in large language models (LLMs) and agentic systems have shown exceptional decision-making capabilities, revealing significant potential for autonomic finance. Current financial trading agents predominantly simulate anthropomorphic roles that inadvertently introduce emotional bia…

Cited by 0SourceScholar
2026

WALKSAFE: Risk-aware Graph Random Walk with Bi-GRPO for LLM Safety

AAAI 2026technical

Large language models (LLMs) may generate harmful outputs on malicious inputs. Existing safety methods, including prompt engineering and model editing, rely on hand-crafted templates or target-driven parameter modifications, limiting their generalizability in unseen harmful scenarios. Post-training

Cited by 0SourcePDFScholar
2025

AGD: Adversarial Game Defense Against Jailbreak Attacks in Large Language Models

ACL 2025long

LLMs demonstrate remarkable utility but remain vulnerable to jailbreak attacks that aim to elicit harmful responses. Existing defenses, including post-training alignment and prompt engineering, rely on training on safety-annotated datasets and safe prompt templates, struggling with adaptability to o…

2025

Accelerating Block Coordinate Descent for LLM Finetuning via Landscape Expansion

NeurIPS 2025poster

Finetuning large language models (LLMs) is a resource-intensive task for researchers in academia, with memory constraints posing a key bottleneck. A classic optimization method, block coordinate descent (BCD), significantly reduces memory cost by segmenting the trainable parameters into multiple blo…

Cited by 0SourceScholar
2025

Adversarial Domain Prompt Tuning and Generation for Single Domain Generalization

CVPR 2025poster

Single domain generalization (SDG) aims to learn a robust model, which could perform well on many unseen domains while there is only one single domain available for training. One of the promising directions for achieving single-domain generalization is to generate out-of-domain (OOD) training data t…

Cited by 0SourcePDFScholar
2025

An Efficient Pore Annotation Framework for Tight Sandstone Images with Segment Anything Model

ICASSP 2025accepted

Analyzing pore structures in tight sandstone thin sections is pivotal for assessing reservoir quality and predicting hydrocarbon migration, providing critical guidance for the formulation of energy extraction strategies. However, the significant variability in pore morphology, scale, and quantity re…

Cited by 0SourceScholar
2025

Chain-of-Model Learning for Language Model

NeurIPS 2025poster

In this paper, we propose a novel learning paradigm, termed *Chain-of-Model* (CoM), which incorporates the causal relationship into the hidden states of each layer as a chain style. thereby introducing great scaling efficiency in model training and inference flexibility in deployment.We introduce th…

Cited by 0SourceScholar
2025

Correlation-Aware Example Selection for In-Context Learning with Nonsymmetric Determinantal Point Processes

EMNLP 2025

LLMs with in-context learning (ICL) obtain remarkable performance but are sensitive to the quality of ICL examples. Prior works on ICL example selection explored unsupervised heuristic methods and supervised LLM-based methods, but they typically focus on the selection of individual examples and igno

Cited by 0SourcePDFScholar
2025

DPGA-TextSyn: Differentially Private Genetic Algorithm for Synthetic Text Generation

ACL 2025finding

Using large language models (LLMs) has a potential risk of privacy leakage since the data with sensitive information may be used for fine-tuning the LLMs. Differential privacy (DP) provides theoretical guarantees of privacy protection, but its practical application in LLMs still has the problem of p…

2025

DYNTEXT: Semantic-Aware Dynamic Text Sanitization for Privacy-Preserving LLM Inference

ACL 2025finding

LLMs face privacy risks when handling sensitive data. To ensure privacy, researchers use differential privacy (DP) to provide protection by adding noise during LLM training. However, users may be hesitant to share complete data with LLMs. Researchers follow local DP to sanitize the text on the user…

2025

Dovetail: A CPU/GPU Heterogeneous Speculative Decoding for LLM inference

EMNLP 2025

With the continuous advancement in the performance of large language models (LLMs), their demand for computational resources and memory has significantly increased, which poses major challenges for efficient inference on consumer-grade devices and legacy servers. These devices typically feature rela

2025

DreamDistribution: Learning Prompt Distribution for Diverse In-distribution Generation

ICLR 2025poster

The popularization of Text-to-Image (T2I) diffusion models enables the generation of high-quality images from text descriptions. However, generating diverse customized images with reference visual attributes remains challenging. This work focuses on personalizing T2I diffusion models at a more abstr…

Cited by 0SourcePDFScholar
2025

EASYTOOL: Enhancing LLM-based Agents with Concise Tool Instruction

NAACL 2025long

There has been a rising interest in utilizing tools in applications of autonomous agents based on large language models (LLMs) to address intricate real-world tasks. To develop LLMbased agents, it usually requires LLMs to understand many tool functions from different tool documentations. However, th…

2025

EvoAgent: Towards Automatic Multi-Agent Generation via Evolutionary Algorithms

NAACL 2025long

The rise of powerful large language models (LLMs) has spurred a new trend in building LLM-based autonomous agents for solving complex tasks, especially multi-agent systems. Despite the remarkable progress, we notice that existing works are heavily dependent on human-designed frameworks, which greatl…

2025

GCML: Gradient Coherence Guided Meta-Learning for Cross-Domain Emerging Topic Rumor Detection

EMNLP 2025

With the emergence of new topics on social media as sources of rumor propagation, addressing the domain shift between the source and target domain and the target domain samples scarcity remains a crucial task in cross-domain rumor detection. Traditional deep learning-based methods and LLM-based meth

Cited by 0SourcePDFScholar
2025

Habitizing Diffusion Planning for Efficient and Effective Decision Making

ICML 2025poster

Diffusion models have shown great promise in decision-making, also known as diffusion planning. However, the slow inference speeds limit their potential for broader real-world applications. Here, we introduce **Habi**, a general framework that transforms powerful but slow diffusion planning models…

2025

Hippocampal-like Sequential Editing for Continual Knowledge Updates in Large Language Models

NeurIPS 2025poster

Large language models (LLMs) are now pivotal in real-world applications. Model editing has emerged as a promising paradigm for efficiently modifying LLMs without full retraining. However, current editing approaches face significant limitations due to parameter drift, which stems from inconsistencies…

Cited by 0SourceScholar
2025

How Do Large Language Models Understand Graph Patterns? A Benchmark for Graph Pattern Comprehension

ICLR 2025poster

Benchmarking the capabilities and limitations of large language models (LLMs) in graph-related tasks is becoming an increasingly popular and crucial area of research. Recent studies have shown that LLMs exhibit a preliminary ability to understand graph structures and node features. However, the pote…

Cited by 4SourcePDFScholar
2025

LLM-based Rumor Detection via Influence Guided Sample Selection and Game-based Perspective Analysis

ACL 2025long

Rumor detection on social media has become an emerging topic. Traditional deep learning-based methods model rumors based on content, propagation structure, or user behavior, but these approaches are constrained by limited modeling capacity and insufficient training corpora. Recent studies have explo…

Cited by 0SourcePDFScholar
2025

LTOS: Layout-controllable Text-object Synthesis via Adaptive Cross-attention Fusions

ICASSP 2025accepted

Controllable text-to-image generation synthesizes visual text and objects in images with certain conditions. However, existing visual text rendering and layout-to-image generation tasks focus on single modality generation or rendering, leaving yet-to-be-bridged gaps between the approaches correspond…

Cited by 0SourceScholar
2025

Learning to Instruct for Visual Instruction Tuning

NeurIPS 2025poster

We propose L2T, an advancement of visual instruction tuning (VIT). While VIT equips Multimodal LLMs (MLLMs) with promising multimodal capabilities, the current design choices for VIT often result in overfitting and shortcut learning, potentially degrading performance. This gap arises from an overemp…

Cited by 7SourcecodeScholar
2025

MMInference: Accelerating Pre-filling for Long-Context Visual Language Models via Modality-Aware Permutation Sparse Attention

ICML 2025poster

The integration of long-context capabilities with visual understanding unlocks unprecedented potential for Vision Language Models (VLMs). However, the quadratic attention complexity during the pre-filling phase remains a significant obstacle to real-world deployment. To overcome this limitation, we…

Cited by 0SourcePDFScholar
2025

MONTROSE: LLM-driven Monte Carlo Tree Search Self-Refinement for Cross-Domain Rumor Detection

ACL 2025finding

With the emergence of new topics on social media as sources of rumor dissemination, addressing the distribution shifts between source and target domains remains a crucial task in cross-domain rumor detection. Existing feature alignment methods, which aim to reduce the discrepancies between domains,…

2025

Mitigate Position Bias in LLMs via Scaling a Single Hidden States Channel

ACL 2025finding

Long-context language models (LCLMs) can process long context, but still exhibit position bias, also known as “lost in the middle”, which indicates placing key information in the middle of the context will significantly affect performance. To mitigating this, we first explore the micro-level manifes…

Cited by 0SourcePDFScholar
2025

Mitigating Hallucinations in Large Vision-Language Models via DPO: On-Policy Data Hold the Key

CVPR 2025poster

Hallucination remains a major challenge for Large Vision-Language Models (LVLMs). Direct Preference Optimization (DPO) has gained increasing attention as a simple solution to hallucination issues. It directly learns from constructed preference pairs that reflect the severity of hallucinations in res…

2025

NeuroLM: A Universal Multi-task Foundation Model for Bridging the Gap between Language and EEG Signals

ICLR 2025poster

Recent advancements for large-scale pre-training with neural signals such as electroencephalogram (EEG) have shown promising results, significantly boosting the development of brain-computer interfaces (BCIs) and healthcare. However, these pre-trained models often require full fine-tuning on each do…

2025

Omni-DNA: A Genomic Model Supporting Sequence Understanding, Long-context, and Textual Annotation

NeurIPS 2025poster

The interpretation of genomic sequences is crucial for understanding biological processes. To handle the growing volume of DNA sequence data, Genomic Foundation Models (GFMs) have been developed by adapting architectures and training paradigms from Large Language Models (LLMs). Despite their remarka…

Cited by 0SourceScholar
2025

Oracle-MoE: Locality-preserving Routing in the Oracle Space for Memory-constrained Large Language Model Inference

ICML 2025poster

Mixture-of-Experts (MoE) is widely adopted to deploy Large Language Models (LLMs) on edge devices with limited memory budgets. Although MoE is, in theory, an inborn memory-friendly architecture requiring only a few activated experts to reside in the memory for inference, current MoE architectures ca…

Cited by 0SourcePDFScholar
2025

SCBench: A KV Cache-Centric Analysis of Long-Context Methods

ICLR 2025poster

Long-context Large Language Models (LLMs) have enabled numerous downstream applications but also introduced significant challenges related to computational and memory efficiency. To address these challenges, optimizations for long-context inference have been developed, centered around the KV cache.…

Cited by 8SourcePDFScholar
2025

SeCom: On Memory Construction and Retrieval for Personalized Conversational Agents

ICLR 2025poster

To deliver coherent and personalized experiences in long-term conversations, existing approaches typically perform retrieval augmented response generation by constructing memory banks from conversation history at either the turn-level, session-level, or through summarization techniques. In this pape…

Cited by 0SourcePDFScholar
2025

SimSort: A Data-Driven Framework for Spike Sorting by Large-Scale Electrophysiology Simulation

NeurIPS 2025poster

Spike sorting is an essential process in neural recording, which identifies and separates electrical signals from individual neurons recorded by electrodes in the brain, enabling researchers to study how specific neurons communicate and process information. Although there exist a number of spike sor…

Cited by 0SourceScholar
2025

Toward Relative Positional Encoding in Spiking Transformers

NeurIPS 2025spotlight

Spiking neural networks (SNNs) are bio-inspired networks that mimic how neurons in the brain communicate through discrete spikes, which have great potential in various tasks due to their energy efficiency and temporal processing capabilities. SNNs with self-attention mechanisms (spiking Transformers…

Cited by 0SourcecodeScholar
2025

Translating Mental Imaginations into Characters with Codebooks and Dynamics-Enhanced Decoding

ICASSP 2025accepted

Advancements in non-invasive electroencephalogram (EEG)-based Brain-Computer Interface (BCI) technology have enabled communication through brain activity, offering significant potential for individuals with motor impairments. Existing methods for decoding characters or words from EEG recordings eith…

Cited by 0SourceScholar
2025

VisRL: Intention-Driven Visual Perception via Reinforced Reasoning

ICCV 2025poster

Visual understanding is inherently intention-driven--humans selectively focus on different regions of a scene based on their goals. Recent advances in large multimodal models (LMMs) enable flexible expression of such intentions through natural language, allowing queries to guide visual reasoning pro…

2025

What Makes a Good Diffusion Planner for Decision Making?

ICLR 2025spotlight

Diffusion models have recently shown significant potential in solving decision-making problems, particularly in generating behavior plans -- also known as diffusion planning. While numerous studies have demonstrated the impressive performance of diffusion planning, the mechanisms behind the key comp…

2025

Zero-resource Hallucination Detection for Text Generation via Graph-based Contextual Knowledge Triples Modeling

AAAI 2025technical

LLMs obtain remarkable performance but suffer from hallucinations. Most research on detecting hallucination focuses on questions with short and concrete correct answers that are easy to check faithfulness. Hallucination detections for text generation with open-ended answers are more hard. Some resea…

2024

3D Parallelism for Transformers via Integer Programming

ICASSP 2024accepted

Transformer models, such as BERT, GPT, and ViT, have been applied to a wide range of areas in recent years, due to their efficacy. In order to improve the training efficiency of Transformer models, different distributed training approaches have been proposed, like Megatron-LM [8]. However, when mult…

Cited by 0SourceScholar
2024

Advancing Spiking Neural Networks for Sequential Modeling with Central Pattern Generators

NeurIPS 2024spotlight

Spiking neural networks (SNNs) represent a promising approach to developing artificial neural networks that are both energy-efficient and biologically plausible. However, applying SNNs to sequential tasks, such as text classification and time-series forecasting, has been hindered by the challenge of…

2024

Auto-Prox: Training-Free Vision Transformer Architecture Search via Automatic Proxy Discovery

AAAI 2024technical

The substantial success of Vision Transformer (ViT) in computer vision tasks is largely attributed to the architecture design. This underscores the necessity of efficient architecture search for designing better ViTs automatically. As training-based architecture search methods are computationally in…

2024

Can Graph Learning Improve Planning in LLM-based Agents?

NeurIPS 2024poster

Task planning in language agents is emerging as an important research topic alongside the development of large language models (LLMs). It aims to break down complex user requests in natural language into solvable sub-tasks, thereby fulfilling the original requests. In this context, the sub-tasks can…

2024

Context-aware Watermark with Semantic Balanced Green-red Lists for Large Language Models

EMNLP 2024main

Watermarking enables people to determine whether the text is generated by a specific model. It injects a unique signature based on the “green-red” list that can be tracked during detection, where the words in green lists are encouraged to be generated. Recent researchers propose to fix the green/red…

Cited by 4SourcePDFScholar
2024

Disentangled Prompt Representation for Domain Generalization

CVPR 2024poster

Domain Generalization (DG) aims to develop a versatile model capable of performing well on unseen target domains. Recent advancements in pre-trained Visual Foundation Models (VFMs) such as CLIP show significant potential in enhancing the generalization abilities of deep models. Although there is a g…

Cited by 9SourcePDFScholar
2024

EEG2Video: Towards Decoding Dynamic Visual Perception from EEG Signals

NeurIPS 2024poster

Our visual experience in daily life are dominated by dynamic change. Decoding such dynamic information from brain activity can enhance the understanding of the brain’s visual processing system. However, previous studies predominately focus on reconstructing static visual stimuli. In this paper, we e…

Cited by 5SourcePDFScholar
2024

Efficient and Effective Time-Series Forecasting with Spiking Neural Networks

ICML 2024poster

Spiking neural networks (SNNs), inspired by the spiking behavior of biological neurons, provide a unique pathway for capturing the intricacies of temporal data. However, applying SNNs to time-series forecasting is challenging due to difficulties in effective temporal alignment, complexities in encod…

2024

Emancipating Event Extraction from the Constraints of Long-Tailed Distribution Data Utilizing Large Language Models

COLING 2024main

Event Extraction (EE) is a challenging task that aims to extract structural event-related information from unstructured text. Traditional methods for EE depend on manual annotations, which are both expensive and scarce. Furthermore, the existing datasets mostly follow the long-tail distribution, sev…

Cited by 2SourcePDFScholar
2024

Exploring the Inefficiency of Heavy Ball as Momentum Parameter Approaches 1

IJCAI 2024poster

The heavy ball momentum method is a commonly used technique for accelerating training processes in the machine learning community. However, empirical evidence suggests that the convergence of stochastic gradient descent (SGD) with heavy ball may slow down when the momentum hyperparameter approaches…

Cited by 0SourcePDFScholar
2024

Improving Large Language Models in Event Relation Logical Prediction

ACL 2024long

Event relations are crucial for narrative understanding and reasoning. Governed by nuanced logic, event relation extraction (ERE) is a challenging task that demands thorough semantic understanding and rigorous logical reasoning. In this paper, we conduct an in-depth investigation to systematically e…

2024

KC-GenRe: A Knowledge-constrained Generative Re-ranking Method Based on Large Language Models for Knowledge Graph Completion

COLING 2024main

The goal of knowledge graph completion (KGC) is to predict missing facts among entities. Previous methods for KGC re-ranking are mostly built on non-generative language models to obtain the probability of each candidate. Recently, generative large language models (LLMs) have shown outstanding perfor…

2024

LLMs Can Find Mathematical Reasoning Mistakes by Pedagogical Chain-of-Thought

IJCAI 2024poster

Self-correction is emerging as a promising approach to mitigate the issue of hallucination in Large Language Models (LLMs). To facilitate effective self-correction, recent research has proposed mistake detection as its initial step. However, current literature suggests that LLMs often struggle with…

2024

Learning Hierarchical Prompt with Structured Linguistic Knowledge for Vision-Language Models

AAAI 2024technical

Prompt learning has become a prevalent strategy for adapting vision-language foundation models to downstream tasks. As large language models (LLMs) have emerged, recent studies have explored the use of category-related descriptions as input to enhance prompt effectiveness. Nevertheless, conventional…

2024

LongLLMLingua: Accelerating and Enhancing LLMs in Long Context Scenarios via Prompt Compression

ACL 2024long

In long context scenarios, large language models (LLMs) face three main challenges: higher computational cost, performance reduction, and position bias. Research indicates that LLM performance hinges on the density and position of key information in the input prompt. Inspired by these findings, we p…

2024

MInference 1.0: Accelerating Pre-filling for Long-Context LLMs via Dynamic Sparse Attention

NeurIPS 2024spotlight

The computational challenges of Large Language Model (LLM) inference remain a significant barrier to their widespread deployment, especially as prompt lengths continue to increase. Due to the quadratic complexity of the attention computation, it takes 30 minutes for an 8B LLM to process a prompt of…

2024

Once Read is Enough: Domain-specific Pretraining-free Language Models with Cluster-guided Sparse Experts for Long-tail Domain Knowledge

NeurIPS 2024poster

Language models (LMs) only pretrained on a general and massive corpus usually cannot attain satisfying performance on domain-specific downstream tasks, and hence, applying domain-specific pretraining to LMs is a common and indispensable practice. However, domain-specific pretraining can be costly an…

Cited by 0SourcePDFScholar
2024

Online Video Quality Enhancement with Spatial-Temporal Look-up Tables

ECCV 2024poster

"Low latency rates are crucial for online video-based applications, such as video conferencing and cloud gaming, which make improving video quality in online scenarios increasingly important. However, existing quality enhancement methods are limited by slow inference speed and the requirement for te…

2024

POMP: Probability-driven Meta-graph Prompter for LLMs in Low-resource Unsupervised Neural Machine Translation

ACL 2024long

Low-resource languages (LRLs) face challenges in supervised neural machine translation (NMT) due to limited parallel data, prompting research in unsupervised NMT.Unsupervised NMT (UNMT), without requiring ground truth, provides solutions for LRL translations using synthetic pseudo-parallel data and…

2024

Perception of Knowledge Boundary for Large Language Models through Semi-open-ended Question Answering

NeurIPS 2024poster

Large Language Models (LLMs) are widely used for knowledge-seeking purposes yet suffer from hallucinations. The knowledge boundary of an LLM limits its factual understanding, beyond which it may begin to hallucinate. Investigating the perception of LLMs' knowledge boundary is crucial for detecting h…

Cited by 4SourcePDFScholar
2024

Stability and Generalization of Asynchronous SGD: Sharper Bounds Beyond Lipschitz and Smoothness

NeurIPS 2024poster

Asynchronous stochastic gradient descent (ASGD) has evolved into an indispensable optimization algorithm for training modern large-scale distributed machine learning tasks. Therefore, it is imperative to explore the generalization performance of the ASGD algorithm. However, the existing results are…

Cited by 4SourcePDFScholar
2024

TaskBench: Benchmarking Large Language Models for Task Automation

NeurIPS 2024poster

In recent years, the remarkable progress of large language models (LLMs) has sparked interest in task automation, which involves decomposing complex tasks described by user instructions into sub-tasks and invoking external tools to execute them, playing a central role in autonomous agents. However,…

2024

Towards Editing Time Series

NeurIPS 2024poster

Synthesizing time series data is pivotal in modern society, aiding effective decision making and ensuring privacy preservation in various scenarios. Time series are associated with various attributes, including trends, seasonality, and external information such as location. Recent research has predo…

2024

Training-free Multi-objective Diffusion Model for 3D Molecule Generation

ICLR 2024poster

Searching for novel and diverse molecular candidates is a critical undertaking in drug and material discovery. Existing approaches have successfully adapted the diffusion model, the most effective generative model in image generation, to create 1D SMILES strings, 2D chemical graphs, or 3D molecular…

Cited by 10SourcePDFScholar
2024

Two-stage Generative Question Answering on Temporal Knowledge Graph Using Large Language Models

ACL 2024findings

Temporal knowledge graph question answering (TKGQA) poses a significant challenge task, due to the temporal constraints hidden in questions and the answers sought from dynamic structured knowledge. Although large language models (LLMs) have made considerable progress in their reasoning ability over…

Cited by 17SourcePDFScholar
2024

Understanding and Improving Training-free Loss-based Diffusion Guidance

NeurIPS 2024poster

Adding additional guidance to pretrained diffusion models has become an increasingly popular research area, with extensive applications in computer vision, reinforcement learning, and AI for science. Recently, several studies have proposed training-free loss-based guidance by using off-the-shelf net…

2024

Unified Medical Image Pre-training in Language-Guided Common Semantic Space

ECCV 2024poster

"Vision-Language Pre-training (VLP) has shown the merits of analysing medical images. It efficiently learns visual representations by leveraging supervisions in their corresponding reports, and in turn facilitates analysis and interpretation of intricate imaging data. However, such observation is pr…

Cited by 6SourcePDFScholar
2023

A Canonicalization-Enhanced Known Fact-Aware Framework For Open Knowledge Graph Link Prediction

IJCAI 2023poster

Open knowledge graph (OpenKG) link prediction aims to predict missing factual triples in the form of (head noun phrase, relation phrase, tail noun phrase). Since triples are not canonicalized, previous methods either focus on canonicalizing noun phrases (NPs) to reduce graph sparsity, or utilize tex…

2023

A Hierarchical Regression Chain Framework for Affective Vocal Burst Recognition

ICASSP 2023accepted

As a common way of emotion signaling via non-linguistic vocalizations, vocal burst (VB) plays an important role in daily social interaction. Understanding and modeling human vocal bursts are indispensable for developing robust and general artificial intelligence. Exploring computational approaches f…

Cited by 0SourceScholar
2023

Adaptive Policy Learning for Offline-to-Online Reinforcement Learning

AAAI 2023technical

Conventional reinforcement learning (RL) needs an environment to collect fresh data, which is impractical when online interactions are costly. Offline RL provides an alternative solution by directly learning from the previously collected dataset. However, it will yield unsatisfactory performance if…

Cited by 27SourcePDFScholar
2023

CircuitNet: A Generic Neural Network to Realize Universal Circuit Motif Modeling

ICML 2023poster

The successes of artificial neural networks (ANNs) are largely attributed to mimicking the human brain structures. Recent advances in neuroscience revealed that neurons interact with each other through various kinds of connectivity patterns to process information, in which the common connectivity pa…

Cited by 0SourcePDFScholar
2023

ContiFormer: Continuous-Time Transformer for Irregular Time Series Modeling

NeurIPS 2023poster

Modeling continuous-time dynamics on irregular time series is critical to account for data evolution and correlations that occur continuously. Traditional methods including recurrent neural networks or Transformer models leverage inductive bias via powerful neural architectures to capture complex pa…

2023

DMFormer: Closing the gap Between CNN and Vision Transformers

ICASSP 2023accepted

Vision transformers have shown excellent performance in computer vision tasks. As the computation cost of their self-attention mechanism is expensive, recent works tried to replace the self-attention mechanism in vision transformers with convolutional operations, which is more efficient with built-i…

Cited by 0SourceScholar
2023

DaMSTF: Domain Adversarial Learning Enhanced Meta Self-Training for Domain Adaptation

ACL 2023long

Self-training emerges as an important research line on domain adaptation. By taking the model’s prediction as the pseudo labels of the unlabeled data, self-training bootstraps the model with pseudo instances in the target domain. However, the prediction errors of pseudo labels (label noise) challeng…

2023

Data Continuity Matters: Improving Sequence Modeling with Lipschitz Regularizer

ICLR 2023top-25%

Sequence modeling is a core problem in machine learning, and various neural networks have been designed to process different types of sequence data. However, few attempts have been made to understand the inherent data property of sequence data, neglecting the critical factor that may significantly a…

Cited by 4SourcePDFScholar
2023

DiffusionNER: Boundary Diffusion for Named Entity Recognition

ACL 2023long

In this paper, we propose DiffusionNER, which formulates the named entity recognition task as a boundary-denoising diffusion process and thus generates named entities from noisy spans. During training, DiffusionNER gradually adds noises to the golden entity boundaries by a fixed forward diffusion pr…

2023

EA-HAS-Bench: Energy-aware Hyperparameter and Architecture Search Benchmark

ICLR 2023top-25%

The energy consumption for training deep learning models is increasing at an alarming rate due to the growth of training data and model scale, resulting in a negative impact on carbon neutrality. Energy consumption is an especially pressing issue for AutoML algorithms because it usually requires rep…

Cited by 1SourcePDFScholar
2023

Explaining Temporal Graph Models through an Explorer-Navigator Framework

ICLR 2023poster

While GNN explanation has recently received significant attention, existing works are consistently designed for static graphs. Due to the prevalence of temporal graphs, many temporal graph models have been proposed, but explaining their predictions remains to be explored. To bridge the gap, in this…

Cited by 20SourcePDFScholar
2023

GRACE: Gradient-guided Controllable Retrieval for Augmenting Attribute-based Text Generation

ACL 2023findings

Attribute-based generation methods are of growing significance in controlling the generation of large pre-trained language models (PLMs). Existing studies control the generation by (1) finetuning the model with attributes or (2) guiding the inference processing toward control signals while freezing…

2023

GROVE: A Retrieval-augmented Complex Story Generation Framework with A Forest of Evidence

EMNLP 2023long findings

Conditional story generation is significant in human-machine interaction, particularly in producing stories with complex plots. While Large language models (LLMs) perform well on multiple NLP tasks, including story generation, it is challenging to generate stories with both complex and creative plot…

Cited by 0SourceScholar
2023

HuggingGPT: Solving AI Tasks with ChatGPT and its Friends in Hugging Face

NeurIPS 2023poster

Solving complicated AI tasks with different domains and modalities is a key step toward artificial general intelligence. While there are numerous AI models available for various domains and modalities, they cannot handle complicated AI tasks autonomously. Considering large language models (LLMs) hav…

2023

Learning Decomposed Spatial Relations for Multi-Variate Time-Series Modeling

AAAI 2023technical

Modeling multi-variate time-series (MVTS) data is a long-standing research subject and has found wide applications. Recently, there is a surge of interest in modeling spatial relations between variables as graphs, i.e., first learning one static graph for each dataset and then exploiting the graph s…

Cited by 20SourcePDFScholar
2023

Learning Joint Structural and Temporal Contextualized Knowledge Embeddings for Temporal Knowledge Graph Completion

ACL 2023findings

Temporal knowledge graph completion that predicts missing links for incomplete temporal knowledge graphs (TKG) is gaining increasing attention. Most existing works have achieved good results by incorporating time information into static knowledge graph embedding methods. However, they ignore the con…

Cited by 14SourcePDFScholar
2023

Learning Topology-Agnostic EEG Representations with Geometry-Aware Modeling

NeurIPS 2023poster

Large-scale pre-training has shown great potential to enhance models on downstream tasks in vision and language. Developing similar techniques for scalp electroencephalogram (EEG) is suitable since unlabelled data is plentiful. Meanwhile, various sampling channel selections and inherent structural a…

Cited by 31SourcePDFScholar
2023

Leveraging Pretrained Representations With Task-Related Keywords for Alzheimer's Disease Detection

ICASSP 2023accepted

With the global population aging rapidly, Alzheimer’s disease (AD) is particularly prominent in older adults, which has an insidious onset and leads to a gradual, irreversible deterioration in cognitive domains (memory, communication, etc.). Speech-based AD detection opens up the possibility of wide…

Cited by 0SourceScholar
2023

Meta-Tsallis-Entropy Minimization: A New Self-Training Approach for Domain Adaptation on Text Classification

IJCAI 2023poster

Text classification is a fundamental task for natural language processing, and adapting text classification models across domains has broad applications. Self-training generates pseudo-examples from the model's predictions and iteratively trains on the pseudo-examples, i.e., minimizes the loss on…

Cited by 5SourcePDFScholar
2023

Over-parameterized Model Optimization with Polyak-{\L}ojasiewicz Condition

ICLR 2023poster

This work pursues the optimization of over-parameterized deep models for superior training efficiency and test performance. We first theoretically emphasize the importance of two properties of over-parameterized models, i.e., the convergence gap and the generalization gap. Subsequent analyses unveil…

Cited by 2SourcePDFScholar
2023

Progressive Meta-Pooling Learning for Lightweight Image Classification Model

ICASSP 2023accepted

Practical networks for edge devices adopt shallow depth and small convolutional kernels to save memory and computational cost, which leads to a restricted receptive field. Conventional efficient learning methods focus on lightweight convolution designs, ignoring the role of the receptive field in ne…

Cited by 0SourceScholar
2023

RD-NAS: Enhancing One-Shot Supernet Ranking Ability Via Ranking Distillation From Zero-Cost Proxies

ICASSP 2023accepted

Neural architecture search (NAS) has made tremendous progress in the automatic design of effective neural network structures but suffers from a heavy computational burden. One-shot NAS significantly alleviates the burden through weight sharing and improves computational efficiency. Zero-shot NAS fur…

Cited by 0SourceScholar
2023

Rotation-Invariant Transformer for Point Cloud Matching

CVPR 2023poster

The intrinsic rotation invariance lies at the core of matching point clouds with handcrafted descriptors. However, it is widely despised by recent deep matchers that obtain the rotation invariance extrinsically via data augmentation. As the finite number of augmented rotations can never span the con…

2023

SIMPLE: Specialized Model-Sample Matching for Domain Generalization

ICLR 2023poster

In domain generalization (DG), most existing methods aspire to fine-tune a specific pretrained model through novel DG algorithms. In this paper, we propose an alternative direction, i.e., to efficiently leverage a pool of pretrained models without fine-tuning. Through extensive empirical and theoret…

2023

Self-Evolution Learning for Mixup: Enhance Data Augmentation on Few-Shot Text Classification Tasks

EMNLP 2023long main

Text classification tasks often encounter few-shot scenarios with limited labeled data, and addressing data scarcity is crucial. Data augmentation with mixup merges sample pairs to generate new pseudos, which can relieve the data deficiency issue in text classification. However, the quality of pseud…

Cited by 0SourceScholar
2023

Similarity Distribution Based Membership Inference Attack on Person Re-identification

AAAI 2023technical

While person Re-identification (Re-ID) has progressed rapidly due to its wide real-world applications, it also causes severe risks of leaking personal information from training data. Thus, this paper focuses on quantifying this risk by membership inference (MI) attack. Most of the existing MI attack…

2023

Stability-Based Generalization Analysis of the Asynchronous Decentralized SGD

AAAI 2023technical

The generalization ability often determines the success of machine learning algorithms in practice. Therefore, it is of great theoretical and practical importance to understand and bound the generalization error of machine learning algorithms. In this paper, we provide the first generalization resul…

Cited by 19SourcePDFScholar
2023

Towards Inference Efficient Deep Ensemble Learning

AAAI 2023technical

Ensemble methods can deliver surprising performance gains but also bring significantly higher computational costs, e.g., can be up to 2048X in large-scale ensemble tasks. However, we found that the majority of computations in ensemble methods are redundant. For instance, over 77% of samples in CIFAR…

2023

Towards Understanding Omission in Dialogue Summarization

ACL 2023long

Dialogue summarization aims to condense the lengthy dialogue into a concise summary, and has recently achieved significant progress. However, the result of existing methods is still far from satisfactory. Previous works indicated that omission is a major factor in affecting the quality of summarizat…

2023

Train Faster, Perform Better: Modular Adaptive Training in Over-Parameterized Models

NeurIPS 2023poster

Despite their prevalence in deep-learning communities, over-parameterized models convey high demands of computational costs for proper training. This work studies the fine-grained, modular-level learning dynamics of over-parameterized models to attain a more efficient and fruitful training strategy.…

Cited by 3SourcePDFScholar
2023

Unsupervised Video Anomaly Detection For Stereotypical Behaviours in Autism

ICASSP 2023accepted

Monitoring and analyzing stereotypical behaviours is important for early intervention and care taking in Autism Spectrum Disorder (ASD). This paper focuses on automatically detecting stereotypical behaviours with computer vision techniques. Off-the-shelf methods tackle this task by supervised classi…

Cited by 0SourceScholar
2022

Bootstrapped Transformer for Offline Reinforcement Learning

NeurIPS 2022accept

Offline reinforcement learning (RL) aims at learning policies from previously collected static trajectory data without interacting with the real environment. Recent works provide a novel perspective by viewing offline RL as a generic sequence generation problem, adopting sequence models such as Tran…

Cited by 59SourcePDFScholar
2022

Cross-Modal Knowledge Distillation in Multi-Modal Fake News Detection

ICASSP 2022accepted

Since the rapid dissemination of fake news brings a lot of negative effects on real society, automatic fake news detection has attracted increasing attention in recent years. In most circumstances, the fake news detection task is a multimodal problem that consists of textual and visual contents. Man…

Cited by 0SourceScholar
2022

Empathetic and Emotionally Positive Conversation Systems with an Emotion-specific Query-Response Memory

EMNLP 2022finding

Emotional conversation systems generate responses for the input queries considering the speaker’s emotions in a conversation. Existing emotional conversation systems output emotional responses according to either a given emotion or the user’s emotion reflected in the input queries. Following a given…

2022

Finding Global Homophily in Graph Neural Networks When Meeting Heterophily

ICML 2022spotlight

We investigate graph neural networks on graphs with heterophily. Some existing methods amplify a node’s neighborhood with multi-hop neighbors to include more nodes with homophily. However, it is a significant challenge to set personalized neighborhood sizes for different nodes. Further, for other ho…

2022

Finite-Time Analysis of Adaptive Temporal Difference Learning with Deep Neural Networks

NeurIPS 2022accept

Temporal difference (TD) learning with function approximations (linear functions or neural networks) has achieved remarkable empirical success, giving impetus to the development of finite-time analysis. As an accelerated version of TD, the adaptive TD has been proposed and proved to enjoy finite-tim…

Cited by 6SourcePDFScholar
2022

Improving Few-Shot Relation Classification by Prototypical Representation Learning with Definition Text

NAACL 2022findings

Few-shot relation classification is difficult because the few instances available may not represent well the relation patterns. Some existing approaches explored extra information such as relation definition, in addition to the instances, to learn a better relation representation. However, the encod…

2022

Invariant Information Bottleneck for Domain Generalization

AAAI 2022technical

Invariant risk minimization (IRM) has recently emerged as a promising alternative for domain generalization. Nevertheless, the loss function is difficult to optimize for nonlinear classifiers and the original optimization objective could fail when pseudo-invariant features and geometric skews exist.…

2022

Modeling Precursors for Temporal Knowledge Graph Reasoning via Auto-encoder Structure

IJCAI 2022poster

Temporal knowledge graph (TKG) reasoning that infers missing facts in the future is an essential and challenging task. When predicting a future event, there must be a narrative evolutionary process composed of closely related historical facts to support the event's occurrence, namely fact precursors…

Cited by 23SourcePDFScholar
2022

Parameter-free Dynamic Graph Embedding for Link Prediction

NeurIPS 2022accept

Dynamic interaction graphs have been widely adopted to model the evolution of user-item interactions over time. There are two crucial factors when modelling user preferences for link prediction in dynamic interaction graphs: 1) collaborative relationship among users and 2) user personalized interact…

2022

Recursive Disentanglement Network

ICLR 2022poster

Disentangled feature representation is essential for data-efficient learning. The feature space of deep models is inherently compositional. Existing $\beta$-VAE-based methods, which only apply disentanglement regularization to the resulting embedding space of deep models, cannot effectively regulari…

Cited by 4SourcePDFScholar
2022

Reinforcement Learning with Automated Auxiliary Loss Search

NeurIPS 2022accept

A good state representation is crucial to solving complicated reinforcement learning (RL) challenges. Many recent works focus on designing auxiliary losses for learning informative representations. Unfortunately, these handcrafted objectives rely heavily on expert knowledge and may be sub-optimal. I…

Cited by 17SourcePDFScholar
2022

RendNet: Unified 2D/3D Recognizer With Latent Space Rendering

CVPR 2022oral

Vector graphics (VG) have been ubiquitous in our daily life with vast applications in engineering, architecture, designs, etc. The VG recognition process of most existing methods is to first render the VG into raster graphics (RG) and then conduct recognition based on RG formats. However, this proce…

Cited by 4PDFScholar
2022

S2 Reducer: High-Performance Sparse Communication to Accelerate Distributed Deep Learning

ICASSP 2022accepted

Distributed stochastic gradient descent (SGD) approach has been widely used in large-scale deep learning, and the gradient collective method is vital to ensure the training scalability of the distributed deep learning system. Collective communication such as AllReduce has been widely adopted for the…

Cited by 0SourceScholar
2022

Social Bot-Aware Graph Neural Network for Early Rumor Detection

COLING 2022main

Early rumor detection is a key challenging task to prevent rumors from spreading widely. Sociological research shows that social bots’ behavior in the early stage has become the main reason for rumors’ wide spread. However, current models do not explicitly distinguish genuine users from social bots,…

2022

Towards Applicable Reinforcement Learning: Improving the Generalization and Sample Efficiency with Policy Ensemble

IJCAI 2022poster

It is challenging for reinforcement learning (RL) algorithms to succeed in real-world applications. Take financial trading as an example, the market information is noisy yet imperfect and the macroeconomic regulation or other factors may shift between training and evaluation, thus it requires both g…

2022

Transcormer: Transformer for Sentence Scoring with Sliding Language Modeling

NeurIPS 2022accept

Sentence scoring aims at measuring the likelihood score of a sentence and is widely used in many natural language processing scenarios, like reranking, which is to select the best sentence from multiple candidates. Previous works on sentence scoring mainly adopted either causal language modeling (CL…

Cited by 13SourcePDFScholar
2022

VRL3: A Data-Driven Framework for Visual Deep Reinforcement Learning

NeurIPS 2022accept

We propose VRL3, a powerful data-driven framework with a simple design for solving challenging visual deep reinforcement learning (DRL) tasks. We analyze a number of major obstacles in taking a data-driven approach, and present a suite of design principles, novel findings, and critical insights abou…

2022

Variational oracle guiding for reinforcement learning

ICLR 2022poster

How to make intelligent decisions is a central problem in machine learning and artificial intelligence. Despite recent successes of deep reinforcement learning (RL) in various decision making problems, an important but under-explored aspect is how to leverage oracle observation (the information that…

2021

Diversity and Consistency: Exploring Visual Question-Answer Pair Generation

EMNLP 2021finding

Although showing promising values to downstream applications, generating question and answer together is under-explored. In this paper, we introduce a novel task that targets question-answer pair generation from visual images. It requires not only generating diverse question-answer pairs but also ke…

2021

Energy-Based Open-World Uncertainty Modeling for Confidence Calibration

ICCV 2021poster

Confidence calibration is of great importance to ensure the reliability of decisions made by machine learning systems. However, discriminative classifiers based on deep neural networks are often criticized for producing overconfident predictions that fail to reflect the true correctness likelihood o…

Cited by 70PDFScholar
2021

Inertial Proximal Deep Learning Alternating Minimization for Efficient Neutral Network Training

ICASSP 2021accepted

In recent years, the Deep Learning Alternating Minimization (DLAM), which is actually the alternating minimization applied to the penalty form of the deep neutral networks training, has been developed as an alternative algorithm to overcome several drawbacks of Stochastic Gradient Descent (SGD) algo…

Cited by 0SourceScholar
2021

Learning Invariant Representations and Risks for Semi-Supervised Domain Adaptation

CVPR 2021poster

The success of supervised learning crucially hinges on the assumption that training data matches test data, which rarely holds in practice due to potential distribution shift. In light of this, most existing methods for unsupervised domain adaptation focus on achieving domain-invariant representatio…

Cited by 111PDFScholar
2021

Recognizing Vector Graphics without Rasterization

NeurIPS 2021poster

In this paper, we consider a different data format for images: vector graphics. In contrast to raster graphics which are widely used in image recognition, vector graphics can be scaled up or down into any resolution without aliasing or information loss, due to the analytic representation of the prim…

2021

Reinforcement Learning Enhanced Explainer for Graph Neural Networks

NeurIPS 2021poster

Graph neural networks (GNNs) have recently emerged as revolutionary technologies for machine learning tasks on graphs. In GNNs, the graph structure is generally incorporated with node representation via the message passing scheme, making the explanation much more challenging. Given a trained GNN mod…

Cited by 81SourcePDFScholar
2021

Scalable and Explainable 1-Bit Matrix Completion via Graph Signal Learning

AAAI 2021technical

One-bit matrix completion is an important class of positive-unlabeled (PU) learning problems where the observations consist of only positive examples, e.g., in top-N recommender systems. For the first time, we show that 1-bit matrix completion can be formulated as the problem of recovering clean gra…

2020

Attentional Fused Temporal Transformation Network for Video Action Recognition

ICASSP 2020accepted

Effective spatiotemporal feature representation is crucial to the video-based action recognition task. Focusing on discriminate spatiotemporal feature learning, we propose Attentional Fused Temporal Transformation Network (AttnTTN) for action recognition on top of popular Temporal Segment Network (T…

Cited by 0SourceScholar
2019

General Proximal Incremental Aggregated Gradient Algorithms: Better and Novel Results under General Scheme

NeurIPS 2019poster

The incremental aggregated gradient algorithm is popular in network optimization and machine learning research. However, the current convergence results require the objective function to be strongly convex. And the existing convergence rates are also limited to linear convergence. Due to the mathema…

Cited by 19SourcePDFScholar
2019

Iteratively Reweighted Penalty Alternating Minimization Methods with Continuation for Image Deblurring

ICASSP 2019accepted

In this paper, we consider a class of nonconvex problems with linear constraints appearing frequently in the area of image processing. We solve this problem by the penalty method and propose the iteratively reweighted alternating minimization algorithm. To speed up the algorithm, we also apply the c…

Cited by 0SourceScholar
2017

Mixture-Rank Matrix Approximation for Collaborative Filtering

NeurIPS 2017poster

Low-rank matrix approximation (LRMA) methods have achieved excellent accuracy among today's collaborative filtering (CF) methods. In existing LRMA methods, the rank of user/item feature matrices is typically fixed, i.e., the same rank is adopted to describe all users/items. However, our studies show…

Cited by 42SourcePDFScholar
2016

Low-Rank Matrix Approximation with Stability

ICML 2016poster

Low-rank matrix approximation has been widely adopted in machine learning applications with sparse data, such as recommender systems. However, the sparsity of the data, incomplete and noisy, introduces challenges to the algorithm stability – small changes in the training data may significantly chang…