← Search

Xike Xie

20 accepted papers

2026

Optimizing KV Cache Eviction from an Output Perturbation Perspective

ICML 2026poster

Large language models have revolutionized natural language processing but face significant challenges of high storage and runtime costs, due to the transformer architecture's reliance on self-attention, particularly the large KV cache for long-sequence inference. Recent efforts to reduce KV cache si…

Cited by 0SourceScholar
2026

Taming the Fragility of KV Cache Eviction in LLM Inference

ICLR 2026poster

Large language models have revolutionized natural language processing, yet their deployment remains hampered by the substantial memory and runtime overhead of the transformer’s Key-Value cache. To mitigate this, recent methods employ a scoring-aggregation framework to evict unimportant cache entries…

Cited by 0SourcecodeScholar
2025

Ada-KV: Optimizing KV Cache Eviction by Adaptive Budget Allocation for Efficient LLM Inference

NeurIPS 2025poster

Large Language Models have excelled in various domains but face efficiency challenges due to the growing Key-Value (KV) cache required for long-sequence inference. Recent efforts aim to reduce KV cache size by evicting vast non-critical cache elements during runtime while preserving generation quali…

Cited by 0SourcecodeScholar
2025

CritiPrefill: A Segment-wise Criticality-based Approach for Prefilling Acceleration in LLMs

ICASSP 2025accepted

Large language models have achieved notable success across various domains, yet efficient inference is still limited by the quadratic computation complexity of the attention mechanism. The inference consists of prefilling and decoding phases. Although several attempts have been made to accelerate de…

Cited by 0SourceScholar
2025

DuetGraph: Coarse-to-Fine Knowledge Graph Reasoning with Dual-Pathway Global-Local Fusion

NeurIPS 2025poster

Knowledge graphs (KGs) are vital for enabling knowledge reasoning across various domains. Recent KG reasoning methods that integrate both global and local information have achieved promising results. However, existing methods often suffer from score over-smoothing, which blurs the distinction betwee…

Cited by 0SourceScholar
2025

FRAG: A Flexible Modular Framework for Retrieval-Augmented Generation based on Knowledge Graphs

ACL 2025finding

To mitigate the hallucination and knowledge deficiency in large language models (LLMs), Knowledge Graph (KG)-based Retrieval-Augmented Generation (RAG) has shown promising potential by utilizing KGs as an external resource to enhance LLM reasoning.However, existing KG-RAG approaches struggle with a…

2025

GraphInsight: Unlocking Insights in Large Language Models for Graph Structure Understanding

ACL 2025long

Although Large Language Models (LLMs) have demonstrated potential in processing graphs, they struggle with comprehending graphical structure information through prompts of graph description sequences, especially as the graph size increases. We attribute this challenge to the uneven memory performanc…

Cited by 0SourcePDFScholar
2025

Lego Sketch: A Scalable Memory-augmented Neural Network for Sketching Data Streams

ICML 2025poster

Sketches, probabilistic structures for estimating item frequencies in infinite data streams with limited space, are widely used across various domains. Recent studies have shifted the focus from handcrafted sketches to neural sketches, leveraging memory-augmented neural networks (MANNs) to enhance t…

2025

Prototype-based Optimal Transport for Out-of-Distribution Detection

IJCAI 2025

Detecting Out-of-Distribution (OOD) inputs is crucial for improving the reliability of deep neural networks in the real-world deployment. In this paper, inspired by the inherent distribution shift between in-distribution (ID) and OOD data, we propose a novel method that leverages optimal transport t

2025

SkewRoute: Training-Free LLM Routing for Knowledge Graph Retrieval-Augmented Generation via Score Skewness of Retrieved Context

EMNLP 2025

Large language models excel at many tasks but often incur high inference costs during deployment. To mitigate hallucination, many systems use a knowledge graph to enhance retrieval-augmented generation (KG-RAG). However, the large amount of retrieved knowledge contexts increase these inference costs

2024

Attacks on Continual Semantic Segmentation by Perturbing Incremental Samples

AAAI 2024technical

As an essential computer vision task, Continual Semantic Segmentation (CSS) has received a lot of attention. However, security issues regarding this task have not been fully studied. To bridge this gap, we study the problem of attacks in CSS in this paper. We first propose a new task, namely, attack…

Cited by 2SourcePDFScholar
2024

Mayfly: a Neural Data Structure for Graph Stream Summarization

ICLR 2024spotlight

A graph is a structure made up of vertices and edges used to represent complex relationships between entities, while a graph stream is a continuous flow of graph updates that convey evolving relationships between entities. The massive volume and high dynamism of graph streams promote research on dat…

Cited by 4SourcePDFScholar
2024

Out-of-Distribution Detection for Learning-Based Chest X-Ray Diagnosis

ICASSP 2024accepted

Deep learning has shown prominence in chest radiography interpretation, which is critical in evaluating various lung and chest diseases, such as pneumonia, emphysema, and tuberculosis. Deploying machine learning model, it is important to detect out-of-distribution (OOD) inputs, which are distinct fr…

Cited by 0SourceScholar
2024

Partial Optimal Transport Based Out-of-Distribution Detection for Open-Set Semi-Supervised Learning

IJCAI 2024poster

Semi-supervised learning (SSL) is a machine learning paradigm that utilizes both labeled and unlabeled data to enhance the performance of learning tasks. However, SSL methods operate under the assumption that the label spaces of labeled and unlabeled data are identical, which may not hold in open-wo…

2024

Prompt Learning with Extended Kalman Filter for Pre-trained Language Models

IJCAI 2024poster

Prompt learning has gained popularity as a means to leverage the knowledge embedded in pre-trained language models (PLMs) for NLP tasks while using a limited number of trainable parameters. While it has shown promise in tasks like sentiment classification and natural language inference, generating s…

Cited by 1SourcePDFScholar
2024

TIKP: Text-to-Image Knowledge Preservation for Continual Semantic Segmentation

AAAI 2024technical

Continual Semantic Segmentation (CSS) is an emerging trend, where catastrophic forgetting has been a perplexing problem. In this paper, we propose a Text-to-Image Knowledge Preservation (TIKP) framework to address this issue. TIKP applies Text-to-Image techniques to CSS by automatically generating p…

Cited by 5SourcePDFScholar
2023

Meta-Sketch: A Neural Data Structure for Estimating Item Frequencies of Data Streams

AAAI 2023technical

To estimate item frequencies of data streams with limited space, sketches are widely used in real applications, including real-time web analytics, network monitoring, and self-driving. Sketches can be viewed as a model which maps the identifier of a stream item to the corresponding frequency domain.…

2020

Inductive Link Prediction for Nodes Having Only Attribute Information

IJCAI 2020poster

Predicting the link between two nodes is a fundamental problem for graph data analytics. In attributed graphs, both the structure and attribute information can be utilized for link prediction. Most existing studies focus on transductive link prediction where both nodes are already in the graph. Howe…