← Search

Shenji Wan

7 accepted papers

2026

3DCS: Datasets and Benchmark for Evaluating Conformational Sensitivity in Molecular Representations

ICLR 2026poster

Molecular representations (MRs) that capture 3D conformations are critical for applications such as reaction prediction, drug design, and material discovery. Yet despite the rapid development of molecular representation models, there is no comprehensive benchmark to evaluate their treatment of 3D co…

Cited by 0SourcecodeScholar
2026

HilbertA: Hilbert-Curve–Aligned Sparse Attention for 2D Structured Data

ICML 2026poster

Designing sparse attention for 2-dimensional image data in diffusion models and vision-language models requires reconciling spatial locality with hardware-efficient execution, a fundamental trade-off that existing methods struggle to resolve. Prior approaches preserve 2D structure through handcrafte…

Cited by 0SourceScholar
2026

InfoFlow KV: Information-Flow-Aware KV Recomputation for Long Context

ICML 2026poster

Retrieval-augmented generation (RAG) for long-context question answering is bottlenecked by inference-time prefilling over large retrieved contexts. A common strategy is to precompute key–value (KV) caches for individual documents and selectively recompute a small subset of tokens to restore global …

Cited by 0SourceScholar
2026

RapTB: Rooted Absorbed Trajectory Balance with Submodular Replay for Stable Autoregressive GFlowNet Training

ICML 2026poster

Generative Flow Networks (GFlowNets) enable fine-tuning large language models to approximate reward-proportional posteriors, but they remain prone to mode collapse, manifesting as prefix collapse and length bias. We attribute this to two factors: (i) weak credit assignment to early prefixes, and (ii…

Cited by 0SourceScholar
2025

Reveal and Release: Iterative LLM Unlearning with Self-generated Data

EMNLP 2025

Large language model (LLM) unlearning has demonstrated effectiveness in removing the influence of undesirable data (also known as forget data). Existing approaches typically assume full access to the forget dataset, overlooking two key challenges: (1) Forget data is often privacy-sensitive, rare, or

Cited by 0SourcePDFScholar