← Search

Zhiyuan Wang

24 accepted papers

2026

A Linear Expectation Constraint for Selective Prediction and Routing with False-Discovery Control

ICML 2026poster

Foundation models often generate unreliable answers, while heuristic uncertainty estimators fail to fully distinguish correct from incorrect outputs, causing users to accept erroneous answers without statistical guarantees. We address this through the lens of false discovery rate (FDR) control, ensu…

Cited by 0SourceScholar
2026

COIN: Uncertainty-Guarding Selective Question Answering for Foundation Models with Provable Risk Guarantees

AAAI 2026technical

Uncertainty quantification (UQ) in foundation models is crucial for identifying and mitigating hallucinations in automatically generated text. However, heuristic UQ approaches lack statistical guarantees for key metrics such as the false discovery rate (FDR) in selective prediction tasks. Previous

Cited by 0SourcePDFScholar
2026

GPan-LoRA: Gaussian Process Amortized Networks for Bayesian Low-Rank Adaptation in Large Language Models

ICML 2026poster

Principled uncertainty quantification (UQ) is increasingly recognized as essential for trustworthy artificial general intelligence (AGI). Bayesian Low-Rank Adaptation (LoRA) provides a principled mechanism for uncertainty-aware fine-tuning of large language models (LLMs). However, existing technique…

Cited by 0SourceScholar
2026

MAPS: Multi-Agent Personality Shaping for Collaborative Reasoning

AAAI 2026technical

Collaborative reasoning with multiple agents offers the potential for more robust and diverse problem-solving. However, existing approaches often suffer from homogeneous agent behaviors and lack of reflective and rethinking capabilities. We propose Multi-Agent Personality Shaping ((MAPS), a novel fr

Cited by 0SourcePDFScholar
2026

PMSPO: Progressive Matching and Semantic-Aware Policy Optimization for Camouflaged Object Detection

ICML 2026poster

Reinforcement learning-based Multimodal Large Language Models (MLLMs) provide new perspectives for visual grounding, yet face significant challenges in Camouflaged Object Detection (COD) where objects blend seamlessly with backgrounds. This stems primarily from: difficulties in multi-object matching…

Cited by 0SourceScholar
2026

RefChess: Monte-Carlo Move Selection for Zero-Shot Referring Image Segmentation

ICML 2026poster

Recent advances in zero-shot referring image segmentation (RIS), driven by foundation models such as SAM and CLIP, have improved cross-modal alignment between visual regions and natural language expressions. Nevertheless, selecting the correct segmentation proposal remains challenging, as existing m…

Cited by 0SourceScholar
2025

A Plug-and-Play Query Synthesis Active Learning Framework for Neural PDE Solvers

NeurIPS 2025poster

In recent developments in scientific machine learning (SciML), neural surrogate solvers for partial differential equations (PDEs) have become powerful tools for accelerating scientific computation for various science and engineering applications. However, training neural PDE solvers often demands a…

Cited by 0SourceScholar
2025

Harnessing the Computation Redundancy in ViTs to Boost Adversarial Transferability

NeurIPS 2025poster

Vision Transformers (ViTs) have demonstrated impressive performance across a range of applications, including many safety-critical tasks. Many previous studies have observed that adversarial examples crafted on ViTs exhibit higher transferability than those crafted on CNNs, indicating that ViTs c…

Cited by 0SourceScholar
2025

HyKGE: A Hypothesis Knowledge Graph Enhanced RAG Framework for Accurate and Reliable Medical LLMs Responses

ACL 2025long

In this paper, we investigate the retrieval-augmented generation (RAG) based on Knowledge Graphs (KGs) to improve the accuracy and reliability of Large Language Models (LLMs). Recent approaches suffer from insufficient and repetitive knowledge retrieval, tedious and time-consuming query parsing, and…

2025

MMTP: Meta-learning-based Multi-Textual Prompt Tuning for Visual-Language Models

ICASSP 2025accepted

Pre-trained Visual-Language Models (VLMs) have demonstrated powerful performance on various downstream tasks. Recently, many prompt tuning methods represented by Context Optimization (CoOp) have effectively adapted VLMs to few-shot tasks. However, the CoOp-based methods suffer from overfitting to ba…

Cited by 5SourceScholar
2025

PanComplex: Leveraging Complex-Valued Neural Networks for Enhanced Pansharpening

IJCAI 2025

Pansharpening combines panchromatic and low-resolution multispectral images to generate high-resolution multispectral images. Previous studies have explored the connection between pansharpening and the frequency domain, but mostly in the real-valued domain, leaving the complex domain relatively unex

2025

SConU: Selective Conformal Uncertainty in Large Language Models

ACL 2025long

As large language models are increasingly utilized in real-world applications, guarantees of task-specific metrics are essential for their reliable deployment. Previous studies have introduced various criteria of conformal uncertainty grounded in split conformal prediction, which offer user-specifie…

2025

Sample then Identify: A General Framework for Risk Control and Assessment in Multimodal Large Language Models

ICLR 2025spotlight

Multimodal Large Language Models (MLLMs) exhibit promising advancements across various tasks, yet they still encounter significant trustworthiness issues. Prior studies apply Split Conformal Prediction (SCP) in language modeling to construct prediction sets with statistical guarantees. However, thes…

Cited by 6SourcePDFScholar
2025

Winding Number-Guided Edge-Preserving Implicit Neural Representation of CAD Surfaces

ICRA 2025

Implicit surface representations have emerged as a powerful tool for the task of 3D reconstruction due to their excellent performance. Yet, when the normal information cannot be available, the previous methods often lead to unsatisfactory reconstruction results, even failure. To this end, we propose

Cited by 0SourceScholar
2024

ConU: Conformal Uncertainty in Large Language Models with Correctness Coverage Guarantees

EMNLP 2024finding

Uncertainty quantification (UQ) in natural language generation (NLG) tasks remains an open challenge, exacerbated by the closed-source nature of the latest large language models (LLMs). This study investigates applying conformal prediction (CP), which can transform any heuristic uncertainty notion i…

2024

INCPrompt: Task-Aware Incremental Prompting for Rehearsal-Free Class-Incremental Learning

ICASSP 2024accepted

This paper introduces INCPrompt, an innovative continual learning solution that effectively addresses catastrophic forgetting. INCPrompt’s key innovation lies in its use of adaptive key-learner and task-aware prompts that capture task-relevant information. This unique combination encapsulates genera…

Cited by 0SourceScholar
2024

ITAKE: Interactive Unstructured Text Annotation and Knowledge Extraction System with LLMs and ModelOps

ACL 2024system demonstrations

Extracting structured knowledge from unstructured text data has a wide range of application prospects, and a pervasive trend is to develop text annotation tools to help extraction. However, they often encounter issues such as single scenario usage, lack of effective human-machine collaboration, insu…

2024

P2DT: Mitigating Forgetting in Task-Incremental Learning with Progressive Prompt Decision Transformer

ICASSP 2024accepted

Catastrophic forgetting poses a substantial challenge for managing intelligent agents controlled by a large model, causing performance degradation when these agents face new tasks. In our work, we propose a novel solution - the Progressive Prompt Decision Transformer (P2DT). This method enhances a t…

Cited by 0SourceScholar
2023

DyCVAE: Learning Dynamic Causal Factors for Non-stationary Series Domain Generalization (Student Abstract)

AAAI 2023technical

Learning domain-invariant representations is a major task of out-of-distribution generalization. To address this issue, recent efforts have taken into accounting causality, aiming at learning the causal factors with regard to tasks. However, extending existing generalization methods for adapting non…

Cited by 0SourcePDFScholar
2023

Learning Dynamic Temporal Relations with Continuous Graph for Multivariate Time Series Forecasting (Student Abstract)

AAAI 2023technical

The recent advance in graph neural networks (GNNs) has inspired a few studies to leverage the dependencies of variables for time series prediction. Despite the promising results, existing GNN-based models cannot capture the global dynamic relations between variables owing to the inherent limitation…

Cited by 4SourcePDFScholar
2023

VecoCare: Visit Sequences-Clinical Notes Joint Learning for Diagnosis Prediction in Healthcare Data

IJCAI 2023poster

Due to the insufficiency of electronic health records (EHR) data utilized in practical diagnosis prediction scenarios, most works are devoted to learning powerful patient representations either from structured EHR data (e.g., temporal medical events, lab test results, etc.) or unstructured data (e.g…

2022

Learning Latent Seasonal-Trend Representations for Time Series Forecasting

NeurIPS 2022accept

Forecasting complex time series is ubiquitous and vital in a range of applications but challenging. Recent advances endeavor to achieve progress by incorporating various deep learning techniques (e.g., RNN and Transformer) into sequential models. However, clear patterns are still hard to extract sin…

Cited by 83SourcePDFScholar
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