← Search

Yankai Chen

21 accepted papers

2026

ConSurv: Multimodal Continual Learning for Survival Analysis

AAAI 2026technical

Survival prediction of cancers is crucial for clinical practice, as it informs mortality risks and influences treatment plans. However, a static model trained on a single dataset fails to adapt to the dynamically evolving clinical environment and continuous data streams, limiting its practical utili

Cited by 0SourcePDFScholar
2026

Distributionally Robust Set Representation Learning Under Inference-Time Element Corruption

ICML 2026poster

Standard Set Representation Learning methods typically excel on curated data but often overlook the challenge of Inference-time Element Corruption. This refers to scenarios where deployed models encounter element-level degradations, such as outliers or missing components, that may distort the set re…

Cited by 0SourceScholar
2026

MUSE: Model-Agnostic Tabular Watermarking via Multi-Sample Selection

ICLR 2026poster

We introduce MUSE, a novel watermarking paradigm for tabular generative models. Existing approaches often exploit DDIM invertibility to watermark tabular diffusion models, but tabular diffusion models suffer from poor invertibility, leading to degraded performance. To overcome this limitation, we le…

Cited by 0SourcecodeScholar
2026

Pedagogically-Inspired Data Synthesis for Language Model Knowledge Distillation

ICLR 2026poster

Knowledge distillation from Large Language Models (LLMs) to smaller models has emerged as a critical technique for deploying efficient AI systems. However, current methods for distillation via synthetic data lack pedagogical awareness, treating knowledge transfer as a one-off data synthesis and trai…

Cited by 0SourceScholar
2026

RECODE: A Benchmark for Research Code DEvelopment with Interactive Human Feedback

ICLR 2026poster

Large language models (LLMs) show the promise in supporting scientific research implementation, yet their ability to generate correct and executable code remains limited. Existing works largely adopt one-shot settings, ignoring the iterative and feedback-driven nature of realistic workflows of scien…

Cited by 0SourcecodeScholar
2026

Search-R2: Enhancing Search-Integrated Reasoning via Actor-Refiner Collaboration

ICML 2026poster

Search-integrated reasoning enables language agents to transcend static parametric knowledge by actively querying external sources. However, training these agents via reinforcement learning is hindered by the *multi-scale credit assignment* problem: existing methods typically rely on sparse, traject…

Cited by 0SourceScholar
2026

SeesawNet: Towards Non-stationary Time Series Forecasting with Balanced Modeling of Common and Specific Dependencies

IJCAI 2026

Instance normalization (IN) is widely used in non-stationary multivariate time series forecasting to reduce distribution shifts and highlight common patterns across samples. However, IN can over-smooth instance-specific structural information that is essential for modeling temporal and cross-channel

Cited by 0Scholar
2026

Set Representation Auxiliary Learning with Adversarial Encoding Perturbation and Optimization

ICLR 2026poster

Sets are a fundamental data structure, and learning their vectorized representations is crucial for many computational problems. Existing methods typically focus on intra-set properties such as permutation invariance and cardinality independence. While effective at preserving basic intra-set semanti…

Cited by 0SourceScholar
2026

S²Drug: Bridging Protein Sequence and 3D Structure in Contrastive Representation Learning for Virtual Screening

AAAI 2026technical

Virtual screening (VS) is an essential task in drug discovery, focusing on the identification of small-molecule ligands that bind to specific protein pockets. Existing deep learning methods, from early regression models to recent contrastive learning approaches, primarily rely on structural data whi

Cited by 0SourcePDFScholar
2025

A Survey of RAG-Reasoning Systems in Large Language Models

EMNLP 2025

Retrieval-Augmented Generation (RAG) lifts the factuality of Large Language Models (LLMs) by injecting external knowledge, yet it falls short on problems that demand multi-step inference; conversely, purely reasoning-oriented approaches often hallucinate or mis-ground facts. This survey synthesizes

Cited by 0SourcePDFScholar
2025

AdmTree: Compressing Lengthy Context with Adaptive Semantic Trees

NeurIPS 2025poster

The quadratic complexity of self-attention limits Large Language Models (LLMs) in processing long contexts, a capability vital for many advanced applications. Context compression aims to mitigate this computational barrier while preserving essential semantic information. However, existing methods of…

Cited by 0SourceScholar
2025

Embracing Trustworthy Brain-Agent Collaboration as Paradigm Extension for Intelligent Assistive Technologies

NeurIPS 2025poster

Brain-Computer Interfaces (BCIs) offer a direct communication pathway between the human brain and external devices, holding significant promise for individuals with severe neurological impairments. However, their widespread adoption is hindered by critical limitations, such as low information transf…

Cited by 0SourceScholar
2025

Multi-Agent Autonomous Driving Systems with Large Language Models: A Survey of Recent Advances, Resources, and Future Directions

EMNLP 2025

Autonomous Driving Systems (ADSs) are revolutionizing transportation by reducing human intervention, improving operational efficiency, and enhancing safety. Large Language Models (LLMs), known for their exceptional planning and reasoning capabilities, have been integrated into ADSs to assist with dr

2025

Teaching According to Talents! Instruction Tuning LLMs with Competence-Aware Curriculum Learning

EMNLP 2025

Efficient instruction tuning aims to enhance the ultimate performance of large language models (LLMs) trained on a given instruction dataset. Curriculum learning as a typical data organization strategy has shown preliminary effectiveness in instruction tuning. However, current curriculum tuning meth

2025

TestNUC: Enhancing Test-Time Computing Approaches and Scaling through Neighboring Unlabeled Data Consistency

ACL 2025long

Test-time computing approaches, which leverage additional computational resources during inference, have been proven effective in enhancing large language model performance. This work introduces a novel, linearly scaling approach, TestNUC, that improves test-time predictions by leveraging the local…

2024

Deep Structural Knowledge Exploitation and Synergy for Estimating Node Importance Value on Heterogeneous Information Networks

AAAI 2024technical

The classic problem of node importance estimation has been conventionally studied with homogeneous network topology analysis. To deal with practical network heterogeneity, a few recent methods employ graph neural models to automatically learn diverse sources of information. However, the major concer…

Cited by 11SourcePDFScholar
2024

HiHPQ: Hierarchical Hyperbolic Product Quantization for Unsupervised Image Retrieval

AAAI 2024technical

Existing unsupervised deep product quantization methods primarily aim for the increased similarity between different views of the identical image, whereas the delicate multi-level semantic similarities preserved between images are overlooked. Moreover, these methods predominantly focus on the Euclid…

2024

Influential Exemplar Replay for Incremental Learning in Recommender Systems

AAAI 2024technical

Personalized recommender systems have found widespread applications for effective information filtering. Conventional models engage in knowledge mining within the static setting to reconstruct singular historical data. Nonetheless, the dynamics of real-world environments are in a constant state of f…

Cited by 12SourcePDFScholar
2023

Hyperbolic Representation Learning: Revisiting and Advancing

ICML 2023poster

The non-Euclidean geometry of hyperbolic spaces has recently garnered considerable attention in the realm of representation learning. Current endeavors in hyperbolic representation largely presuppose that the underlying hierarchies can be automatically inferred and preserved through the adaptive opt…

2023

Mitigating the Popularity Bias of Graph Collaborative Filtering: A Dimensional Collapse Perspective

NeurIPS 2023spotlight

Graph-based Collaborative Filtering (GCF) is widely used in personalized recommendation systems. However, GCF suffers from a fundamental problem where features tend to occupy the embedding space inefficiently (by spanning only a low-dimensional subspace). Such an effect is characterized in GCF by th…

Cited by 27SourcePDFScholar
2020

Efficient Community Search over Large Directed Graph: An Augmented Index-based Approach

IJCAI 2020poster

Given a graph G and a query vertex q, the topic of community search (CS), aiming to retrieve a dense subgraph of G containing q, has gained much attention. Most existing works focus on undirected graphs which overlooks the rich information carried by the edge directions. Recently, the problem of com…

Cited by 0SourcePDFScholar