← Search

Aosong Feng

19 accepted papers

2026

CSRv2: Unlocking Ultra-Sparse Embeddings

ICLR 2026poster

In the era of large foundation models, the quality of embeddings has become a central determinant of downstream task performance and overall system capability. Yet widely used dense embeddings are often extremely high-dimensional (e.g., 4096), incurring substantial costs in storage, memory, and inf…

Cited by 0SourcecodeScholar
2026

Efficient Test-Time Scaling via Hierarchical Search and Self-Verification for Discrete Diffusion Language Models

ICML 2026poster

Inference-time compute has re-emerged as a practical way to improve LLM reasoning. Most test-time scaling (TTS) algorithms rely on autoregressive decoding, which is ill-suited to discrete diffusion language models (dLLMs) due to their parallel decoding over the entire sequence. As a result, developi…

Cited by 0SourceScholar
2026

No More K-means: Single-Stage Sparse Coding for Efficient Multi-Vector Retrieval

ICML 2026poster

Multi-vector retrieval (MVR) models, exemplified by ColBERT, have established new benchmarks in retrieval accuracy by preserving fine-grained token-level interactions. However, this granularity imposes prohibitive storage and retrieval efficiency bottlenecks: to manage the immense memory footprint a…

Cited by 0SourceScholar
2026

Threshold-Guided Optimization for Visual Generative Models

ICML 2026poster

Aligning large visual generative models with human feedback is often performed through pairwise preference optimization. While such approaches are conceptually simple, they fundamentally rely on annotated pairs, limiting scalability in settings where feedback is collected as independent scalar ratin…

Cited by 0SourceScholar
2025

An Item Is Worth a Prompt: Versatile Image Editing with Disentangled Control

AAAI 2025technical

Building on the success of text-to-image diffusion models (DPMs), image editing is an important application to enable human interaction with AI-generated content. Among various editing methods, editing within the prompt space gains more attention due to its capacity and simplicity of controlling sem…

Cited by 6SourcePDFScholar
2025

Hyperbolic Fine-Tuning for Large Language Models

NeurIPS 2025spotlight

Large language models (LLMs) have demonstrated remarkable performance on various tasks. However, it remains an open question whether the default Euclidean space is the most suitable choice for embedding tokens in LLMs. In this study, we investigate the non-Euclidean characteristics of LLMs. O…

Cited by 0SourcecodeScholar
2025

MMLU-ProX: A Multilingual Benchmark for Advanced Large Language Model Evaluation

EMNLP 2025

Existing large language model (LLM) evaluation benchmarks primarily focus on English, while current multilingual tasks lack parallel questions that specifically assess cross-lingual reasoning abilities. This dual limitation makes it challenging to assess LLMs’ performance in the multilingual setting

Cited by 0SourcePDFScholar
2025

MindLLM: A Subject-Agnostic and Versatile Model for fMRI-to-text Decoding

ICML 2025poster

Decoding functional magnetic resonance imaging (fMRI) signals into text has been a key challenge in the neuroscience community, with the potential to advance brain-computer interfaces and uncover deeper insights into brain mechanisms. However, existing approaches often struggle with suboptimal predi…

Cited by 0SourcePDFScholar
2025

SafeKey: Amplifying Aha-Moment Insights for Safety Reasoning

EMNLP 2025

Large Reasoning Models (LRMs) introduce a new generation paradigm of explicitly reasoning before answering, leading to remarkable improvements in complex tasks. However, they pose great safety risks against harmful queries and adversarial attacks. While recent mainstream safety efforts on LRMs, supe

2025

TRACE: Grounding Time Series in Context for Multimodal Embedding and Retrieval

NeurIPS 2025poster

The ubiquity of dynamic data in domains such as weather, healthcare, and energy underscores a growing need for effective interpretation and retrieval of time-series data. These data are inherently tied to domain-specific contexts, such as clinical notes or weather narratives, making cross-modal retr…

Cited by 0SourceScholar
2024

DTGB: A Comprehensive Benchmark for Dynamic Text-Attributed Graphs

NeurIPS 2024poster

Dynamic text-attributed graphs (DyTAGs) are prevalent in various real-world scenarios, where each node and edge are associated with text descriptions, and both the graph structure and text descriptions evolve over time. Despite their broad applicability, there is a notable scarcity of benchmark data…

2024

From Similarity to Superiority: Channel Clustering for Time Series Forecasting

NeurIPS 2024poster

Time series forecasting has attracted significant attention in recent decades. Previous studies have demonstrated that the Channel-Independent (CI) strategy improves forecasting performance by treating different channels individually, while it leads to poor generalization on unseen instances and…

2024

Integrating View Conditions for Image Synthesis

IJCAI 2024poster

In the field of image processing, applying intricate semantic modifications within existing images remains an enduring challenge. This paper introduces a pioneering framework that integrates viewpoint information to enhance the control of image editing tasks, especially for interior design scenes. B…

2024

Long Sequence Modeling with Attention Tensorization: From Sequence to Tensor Learning

EMNLP 2024finding

As the demand for processing extended textual data grows, the ability to handle long-range dependencies and maintain computational efficiency is more critical than ever. One of the key issues for long-sequence modeling using attention-based model is the mismatch between the limited-range modeling po…

Cited by 2SourcePDFScholar
2024

Solving General Noisy Inverse Problem via Posterior Sampling: A Policy Gradient Viewpoint

AISTATS 2024poster

Solving image inverse problems (e.g., super-resolution and inpainting) requires generating a high fidelity image that matches the given input (the low-resolution image or the masked image). By using the input image as guidance, we can leverage a pretrained diffusion generative model to solve a wide…

2023

Diffuser: Efficient Transformers with Multi-Hop Attention Diffusion for Long Sequences

AAAI 2023technical

Efficient Transformers have been developed for long sequence modeling, due to their subquadratic memory and time complexity. Sparse Transformer is a popular approach to improving the efficiency of Transformers by restricting self-attention to locations specified by the predefined sparse patterns. Ho…

2023

HiPool: Modeling Long Documents Using Graph Neural Networks

ACL 2023short

Encoding long sequences in Natural Language Processing (NLP) is a challenging problem. Though recent pretraining language models achieve satisfying performances in many NLP tasks, they are still restricted by a pre-defined maximum length, making them challenging to be extended to longer sequences. S…

2022

KerGNNs: Interpretable Graph Neural Networks with Graph Kernels

AAAI 2022technical

Graph kernels are historically the most widely-used technique for graph classification tasks. However, these methods suffer from limited performance because of the hand-crafted combinatorial features of graphs. In recent years, graph neural networks (GNNs) have become the state-of-the-art method in…