← Search

Yifei Sun

16 accepted papers

2026

BRAIN-HGCN: A HYPERBOLIC GRAPH CONVOLUTIONAL NETWORK FOR BRAIN FUNCTIONAL NETWORK ANALYSIS

ICASSP 2026oral

Functional magnetic resonance imaging (fMRI) reveals complex brain functional networks with hierarchical topologies crucial for cognitive processing. Standard Euclidean Graph Neural Networks (GNNs) often struggle to represent these hierarchical structures without high distortion due to inherent spat…

Cited by 0SourcePDFScholar
2026

Can we generate portable representations for clinical time series data using LLMs?

ICLR 2026poster

Deploying clinical ML is slow and brittle: models that work at one hospital often degrade under distribution shifts at the next. In this work, we study a simple question -- can large language models (LLMs) create portable patient embeddings i.e. representations of patients enable a downstream predic…

Cited by 0SourceScholar
2026

GEODESIC PROTOTYPE MATCHING VIA DIFFUSION MAPS FOR INTERPRETABLE FINE-GRAINED RECOGNITION

ICASSP 2026oral

Nonlinear manifolds are pervasive in deep visual features, where Euclidean distances can misrepresent true similarity. This mismatch is particularly detrimental to prototype-based interpretable fine-grained recognition, where even subtle semantic distinctions are crucial. To mitigate this issue, thi…

Cited by 0SourcePDFScholar
2026

WDT-MD: Wavelet Diffusion Transformers for Microaneurysm Detection in Fundus Images

AAAI 2026technical

Microaneurysms (MAs), the earliest pathognomonic signs of Diabetic Retinopathy (DR), present as sub-60 μm lesions in fundus images with highly variable photometric and morphological characteristics, rendering manual screening not only labor-intensive but inherently error-prone. While diffusion-based

Cited by 2SourcePDFScholar
2025

Multi-Label Node Classification with Label Influence Propagation

ICLR 2025poster

Graphs are a complex and versatile data structure used across various domains, with possibly multi-label nodes playing a particularly crucial role. Examples include proteins in PPI networks with multiple functions and users in social or e-commerce networks exhibiting diverse interests. Tackling mu…

Cited by 0SourcePDFScholar
2025

Robots Pre-train Robots: Manipulation-Centric Robotic Representation from Large-Scale Robot Datasets

ICLR 2025poster

The pre-training of visual representations has enhanced the efficiency of robot learning. Due to the lack of large-scale in-domain robotic datasets, prior works utilize in-the-wild human videos to pre-train robotic visual representation. Despite their promising results, representations from human vi…

2024

AfriInstruct: Instruction Tuning of African Languages for Diverse Tasks

EMNLP 2024finding

Large language models (LLMs) for African languages perform worse compared to their performance in high-resource languages. To address this issue, we introduce AfriInstruct, which specializes in instruction-tuning of multiple African languages covering various tasks. We trained the LLaMa-2-7B using c…

2024

CGS-Mask: Making Time Series Predictions Intuitive for All

AAAI 2024technical

Artificial intelligence (AI) has immense potential in time series prediction, but most explainable tools have limited capabilities in providing a systematic understanding of important features over time. These tools typically rely on evaluating a single time point, overlook the time ordering of inpu…

Cited by 1SourcePDFScholar
2024

Exploring Correlations of Self-Supervised Tasks for Graphs

ICML 2024poster

Graph self-supervised learning has sparked a research surge in training informative representations without accessing any labeled data. However, our understanding of graph self-supervised learning remains limited, and the inherent relationships between various self-supervised tasks are still unexplo…

2024

Fine-Tuning Graph Neural Networks by Preserving Graph Generative Patterns

AAAI 2024technical

Recently, the paradigm of pre-training and fine-tuning graph neural networks has been intensively studied and applied in a wide range of graph mining tasks. Its success is generally attributed to the structural consistency between pre-training and downstream datasets, which, however, does not hold…

2024

G-Retriever: Retrieval-Augmented Generation for Textual Graph Understanding and Question Answering

NeurIPS 2024poster

Given a graph with textual attributes, we enable users to `chat with their graph': that is, to ask questions about the graph using a conversational interface. In response to a user's questions, our method provides textual replies and highlights the relevant parts of the graph. While existing works i…

2023

A Composite Multi-Attention Framework for Intraoperative Hypotension Early Warning

AAAI 2023technical

Intraoperative hypotension (IOH) events warning plays a crucial role in preventing postoperative complications, such as postoperative delirium and mortality. Despite significant efforts, two fundamental problems limit its wide clinical use. The well-established IOH event warning systems are often bu…

Cited by 5SourcePDFScholar
2023

ES-Mask: Evolutionary Strip Mask for Explaining Time Series Prediction (Student Abstract)

AAAI 2023technical

Machine learning models are increasingly used in time series prediction with promising results. The model explanation of time series prediction falls behind the model development and makes less sense to users in understanding model decisions. This paper proposes ES-Mask, a post-hoc and model-agnosti…

Cited by 1SourcePDFScholar
2023

Towards Learning to Discover Money Laundering Sub-network in Massive Transaction Network

AAAI 2023technical

Anti-money laundering (AML) systems play a critical role in safeguarding global economy. As money laundering is considered as one of the top group crimes, there is a crucial need to discover money laundering sub-network behind a particular money laundering transaction for a robust AML system. Howeve…

Cited by 17SourcePDFScholar
2022

Beyond Homophily: Structure-aware Path Aggregation Graph Neural Network

IJCAI 2022poster

Graph neural networks (GNNs) have been intensively studied in various real-world tasks. However, the homophily assumption of GNNs' aggregation function limits their representation learning ability in heterophily graphs. In this paper, we shed light on the path level patterns in graphs that can exp…