← Search

Bryan Hooi

86 accepted papers

2026

AliMark: Enhancing Robustness of Sentence-Level Watermarks Against Text Paraphrasing

ICML 2026poster

Existing sentence-level watermarking methods enhance robustness to paraphrasing by anchoring watermarks in sentence semantics. However, their prefix-based designs remain vulnerable to structural perturbations, such as sentence splitting and merging, which commonly arise under strong paraphrasers lik…

Cited by 0SourceScholar
2026

Conversation for Non-verifiable Learning: Self-Evolving Large Language Models through Meta-Evaluation

ICML 2026poster

Training large language models (LLMs) for non-verifiable tasks—such as creative writing, dialogue, and ethical reasoning—remains challenging due to the absence of ground-truth labels. While LLM-as-Judge approaches offer a scalable alternative to human feedback, they face a fundamental limitation: pe…

Cited by 0SourceScholar
2026

DGP: A Dual-Granularity Prompting Framework for Fraud Detection with Graph-Enhanced LLMs

AAAI 2026technical

Real-world fraud detection applications benefit from graph learning techniques that jointly exploit node features—often rich in textual data—and graph structural information. Recently, Graph-Enhanced LLMs have emerged as a promising graph learning approach that converts graph information into prompt

Cited by 0SourcePDFScholar
2026

Echoless Label-Based Pre-computation for Memory-Efficient Heterogeneous Graph Learning

AAAI 2026technical

Heterogeneous Graph Neural Networks (HGNNs) are widely used for deep learning on heterogeneous graphs. Typical end-to-end HGNNs require repetitive message passing during training, limiting efficiency for large-scale real-world graphs. Pre-computation-based HGNNs address this by performing message pa

Cited by 0SourcePDFScholar
2026

EvoTest: Evolutionary Test-Time Learning for Self-Improving Agentic Systems

ICLR 2026poster

A fundamental limitation of current AI agents is their inability to learn complex skills on the fly at test time, often behaving like “clever but clueless interns” in novel environments. This severely limits their practical utility. To systematically measure and drive progress on this challenge, we…

Cited by 0SourcecodeScholar
2026

Just-In-Time Reinforcement Learning: Continual Learning in LLM Agents Without Gradient Updates

ICML 2026spotlight

While Large Language Model (LLM) agents excel at general tasks, they inherently struggle with continual adaptation due to the frozen weights after deployment. Conventional reinforcement learning (RL) offers a solution but incurs prohibitive computational costs and the risk of catastrophic forgetting…

Cited by 0SourceScholar
2026

NTSFormer: A Self-Teaching Graph Transformer for Multimodal Isolated Cold-Start Node Classification

AAAI 2026technical

Isolated cold-start node classification on multimodal graphs is challenging because such nodes have no edges and often have missing modalities (e.g., absent text or image features). Existing methods address structural isolation by degrading graph learning models to multilayer perceptrons (MLPs) for

Cited by 0SourcePDFScholar
2026

Seeing Through Deception: Uncovering Misleading Creator Intent in Multimodal News with Vision-Language Models

ICLR 2026poster

The impact of misinformation arises not only from factual inaccuracies but also from the misleading narratives that creators deliberately embed. Interpreting such creator intent is therefore essential for multimodal misinformation detection (MMD) and effective information governance. To this end, we…

Cited by 0SourcecodeScholar
2026

TrustGen: A Platform of Dynamic Benchmarking on the Trustworthiness of Generative Foundation Models

ICLR 2026poster

Generative foundation models (GenFMs), such as large language models and text-to-image systems, have demonstrated remarkable capabilities in various downstream applications. As they are increasingly deployed in high-stakes applications, assessing their trustworthiness has become both a critical nece…

Cited by 0SourceScholar
2026

Turning Bias into Bugs: Bandit-Guided Style Manipulation Attacks on LLM Judges

ICML 2026poster

Large Language Models (LLMs) are increasingly employed as automated judges for evaluating generative models. However, their known stylistic biases, such as a preference for verbosity or specific sentence structures, present an underexplored security vulnerability. In this work, we introduce BITE (BI…

Cited by 0SourceScholar
2026

VPI-Bench: Visual Prompt Injection Attacks for Computer-Use Agents

ICLR 2026poster

Computer-Use Agents (CUAs) with full system access enable powerful task automation but pose significant security and privacy risks due to their ability to manipulate files, access user data, and execute arbitrary commands. While prior work has focused on browser-based agents and HTML-level attacks,…

Cited by 0SourcecodeScholar
2025

Adapting Precomputed Features for Efficient Graph Condensation

ICML 2025poster

Graph Neural Networks (GNNs) face significant computational challenges when handling large-scale graphs. To address this, Graph Condensation (GC) methods aim to compress large graphs into smaller, synthetic ones that are more manageable for GNN training. Recently, trajectory matching methods have sh…

2025

Automating Steering for Safe Multimodal Large Language Models

EMNLP 2025

Recent progress in Multimodal Large Language Models (MLLMs) has unlocked powerful cross-modal reasoning abilities, but also raised new safety concerns, particularly when faced with adversarial multimodal inputs. To improve the safety of MLLMs during inference, we introduce a modular and adaptive inf

2025

Backdoor-Powered Prompt Injection Attacks Nullify Defense Methods

EMNLP 2025

With the development of technology, large language models (LLMs) have dominated the downstream natural language processing (NLP) tasks. However, because of the LLMs’ instruction-following abilities and inability to distinguish the instructions in the data content, such as web pages from search engin

2025

CHiP: Cross-modal Hierarchical Direct Preference Optimization for Multimodal LLMs

ICLR 2025poster

Multimodal Large Language Models (MLLMs) still struggle with hallucinations despite their impressive capabilities. Recent studies have attempted to mitigate this by applying Direct Preference Optimization (DPO) to multimodal scenarios using preference pairs from text-based responses. However, our an…

2025

Can Indirect Prompt Injection Attacks Be Detected and Removed?

ACL 2025long

Prompt injection attacks manipulate large language models (LLMs) by misleading them to deviate from the original input instructions and execute maliciously injected instructions, because of their instruction-following capabilities and inability to distinguish between the original input instructions…

2025

Can Knowledge Graphs Make Large Language Models More Trustworthy? An Empirical Study Over Open-ended Question Answering

ACL 2025long

Recent works integrating Knowledge Graphs (KGs) have shown promising improvements in enhancing the reasoning capabilities of Large Language Models (LLMs). However, existing benchmarks primarily focus on closed-ended tasks, leaving a gap in evaluating performance on more complex, real-world scenarios…

Cited by 0SourcePDFScholar
2025

Con-ReCall: Detecting Pre-training Data in LLMs via Contrastive Decoding

COLING 2025main

The training data in large language models is key to their success, but it also presents privacy and security risks, as it may contain sensitive information. Detecting pre-training data is crucial for mitigating these concerns. Existing methods typically analyze target text in isolation or solely wi…

2025

ConfTuner: Training Large Language Models to Express Their Confidence Verbally

NeurIPS 2025poster

Large Language Models (LLMs) are increasingly deployed in high-stakes domains such as science, law, and healthcare, where accurate expressions of uncertainty are essential for reliability and trust. However, current LLMs are often observed to generate incorrect answers with high confidence—a phenome…

Cited by 0SourcecodeScholar
2025

DRS: Deep Question Reformulation With Structured Output

ACL 2025finding

Question answering represents a core capability of large language models (LLMs). However, when individuals encounter unfamiliar knowledge in texts, they often formulate questions that the text itself cannot answer due to insufficient understanding of the underlying information. Recent studies reveal…

2025

Defense Against Prompt Injection Attack by Leveraging Attack Techniques

ACL 2025long

With the advancement of technology, large language models (LLMs) have achieved remarkable performance across various natural language processing (NLP) tasks, powering LLM-integrated applications like Microsoft Copilot. However, as LLMs continue to evolve, new vulnerabilities, especially prompt injec…

Cited by 0SourcePDFScholar
2025

Exploring Visual Vulnerabilities via Multi-Loss Adversarial Search for Jailbreaking Vision-Language Models

CVPR 2025poster

Despite inheriting security measures from underlying language models, Vision-Language Models (VLMs) may still be vulnerable to safety alignment issues. Through empirical analysis, we uncover two critical findings: scenario-matched images can significantly amplify harmful outputs, and contrary to com…

Cited by 1SourcePDFScholar
2025

FiDeLiS: Faithful Reasoning in Large Language Models for Knowledge Graph Question Answering

ACL 2025finding

Large Language Models (LLMs) are often challenged by generating erroneous or hallucinated responses, especially in complex reasoning tasks. Leveraging Knowledge Graphs (KGs) as external knowledge sources has emerged as a viable solution. However, existing KG-enhanced methods, either retrieval-based…

2025

FlipAttack: Jailbreak LLMs via Flipping

ICML 2025poster

This paper proposes a simple yet effective jailbreak attack named FlipAttack against black-box LLMs. First, from the autoregressive nature, we reveal that LLMs tend to understand the text from left to right and find that they struggle to comprehend the text when the perturbation is added to the left…

2025

GRAVER: Generative Graph Vocabularies for Robust Graph Foundation Models Fine-tuning

NeurIPS 2025poster

Inspired by the remarkable success of foundation models in language and vision, Graph Foundation Models (GFMs) hold significant promise for broad applicability across diverse graph tasks and domains. However, existing GFMs struggle with unstable few-shot fine-tuning, where both performance and adapt…

Cited by 0SourceScholar
2025

GuardReasoner-VL: Safeguarding VLMs via Reinforced Reasoning

NeurIPS 2025poster

To enhance the safety of VLMs, this paper introduces a novel reasoning-based VLM guard model dubbed GuardReasoner-VL. The core idea is to incentivize the guard model to deliberatively reason before making moderation decisions via online RL. First, we construct GuardReasoner-VLTrain, a reasoning corp…

Cited by 0SourcecodeScholar
2025

How Does Response Length Affect Long-Form Factuality

ACL 2025finding

Large language models (LLMs) are widely used for long-form text generation. However, factual errors in the responses would undermine their reliability. Despite growing attention to LLM factuality, the effect of response length on factuality remains underexplored. In this work, we systematically inve…

2025

How Much Can Transfer? BRIDGE: Bounded Multi-Domain Graph Foundation Model with Generalization Guarantees

ICML 2025poster

Graph Foundation Models hold significant potential for advancing multi-domain graph learning, yet their full capabilities remain largely untapped. Existing works show promising task performance with the “pretrain-then-prompt” paradigm, which lacks theoretical foundations to understand why it works a…

Cited by 0SourcePDFScholar
2025

How to Make Large Language Models Generate 100% Valid Molecules?

EMNLP 2025

Molecule generation is key to drug discovery and materials science, enabling the design of novel compounds with specific properties. Large language models (LLMs) can learn to perform a wide range of tasks from just a few examples. However, generating valid molecules using representations like SMILES

2025

LongRecipe: Recipe for Efficient Long Context Generalization in Large Language Models

ACL 2025long

Large language models (LLMs) face significant challenges in handling long-context tasks because of their limited effective context window size during pretraining, which restricts their ability to generalize over extended sequences. Meanwhile, extending the context window in LLMs through post-pretrai…

2025

MLR-Bench: Evaluating AI Agents on Open-Ended Machine Learning Research

NeurIPS 2025poster

Recent advancements in AI agents have demonstrated their growing potential to drive and support scientific discovery. In this work, we introduce MLR-Bench, a comprehensive benchmark for evaluating AI agents on open-ended machine learning research. MLR-Bench includes three key components: (1) 201 res…

Cited by 0SourcecodeScholar
2025

Making Every Step Effective: Jailbreaking Large Vision-Language Models Through Hierarchical KV Equalization

EMNLP 2025

In the realm of large vision-language models (LVLMs), adversarial jailbreak attacks serve as a red-teaming approach to identify safety vulnerabilities of these models and their associated defense mechanisms. However, we identify a critical limitation: not every adversarial optimization step leads to

Cited by 0SourcePDFScholar
2025

Modality-Independent Graph Neural Networks with Global Transformers for Multimodal Recommendation

AAAI 2025technical

Multimodal recommendation systems can learn users' preferences from existing user-item interactions as well as the semantics of multimodal data associated with items. Many existing methods model this through a multimodal user-item graph, approaching multimodal recommendation as a graph learning task…

2025

Monte Carlo Tree Search for Comprehensive Exploration in LLM-Based Automatic Heuristic Design

ICML 2025poster

Handcrafting heuristics for solving complex optimization tasks (e.g., route planning and task allocation) is a common practice but requires extensive domain knowledge. Recently, Large Language Model (LLM)-based automatic heuristic design (AHD) methods have shown promise in generating high-quality he…

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

PhishAgent: A Robust Multimodal Agent for Phishing Webpage Detection

AAAI 2025technical

Phishing attacks are a major threat to online security, exploiting user vulnerabilities to steal sensitive information. Various methods have been developed to counteract phishing, each with varying levels of accuracy, but they also face notable limitations. In this study, we introduce PhishAgent, a…

Cited by 41SourcePDFScholar
2025

ReLearn: Unlearning via Learning for Large Language Models

ACL 2025long

Current unlearning methods for large language models usually rely on reverse optimization to reduce target token probabilities. However, this paradigm disrupts the subsequent tokens prediction, degrading model performance and linguistic coherence. Moreover, existing evaluation metrics overemphasize…

2025

TopicAttack: An Indirect Prompt Injection Attack via Topic Transition

EMNLP 2025

Large language models (LLMs) have shown remarkable performance across a range of NLP tasks. However, their strong instruction-following capabilities and inability to distinguish instructions from data content make them vulnerable to indirect prompt injection attacks. In such attacks, instructions wi

Cited by 0SourcePDFScholar
2025

Tricking Retrievers with Influential Tokens: An Efficient Black-Box Corpus Poisoning Attack

NAACL 2025long

Retrieval-augmented generation (RAG) systems enhance large language models by incorporating external knowledge, addressing issues like outdated internal knowledge and hallucination. However, their reliance on external knowledge bases makes them vulnerable to corpus poisoning attacks, where adversari…

Cited by 0SourcePDFScholar
2025

Vulnerability of LLMs to Vertically Aligned Text Manipulations

ACL 2025long

Vertical text input is commonly encountered in various real-world applications, such as mathematical computations and word-based Sudoku puzzles. While current large language models (LLMs) have excelled in natural language tasks, they remain vulnerable to variations in text formatting.Recent research…

Cited by 0SourcePDFScholar
2025

Words or Vision: Do Vision-Language Models Have Blind Faith in Text?

CVPR 2025poster

Vision-Language Models (VLMs) excel in integrating visual and textual information for vision-centric tasks, but their handling of inconsistencies between modalities is underexplored. We investigate VLMs' modality preferences when faced with visual data and varied textual inputs in vision-centered se…

2024

$\text{ID}^3$: Identity-Preserving-yet-Diversified Diffusion Models for Synthetic Face Recognition

NeurIPS 2024poster

Synthetic face recognition (SFR) aims to generate synthetic face datasets that mimic the distribution of real face data, which allows for training face recognition models in a privacy-preserving manner. Despite the remarkable potential of diffusion models in image generation, current diffusion-based…

Cited by 4SourcePDFScholar
2024

Can LLMs Express Their Uncertainty? An Empirical Evaluation of Confidence Elicitation in LLMs

ICLR 2024poster

Empowering large language models (LLMs) to accurately express confidence in their answers is essential for reliable and trustworthy decision-making. Previous confidence elicitation methods, which primarily rely on *white-box access* to internal model information or model fine-tuning, have become les…

2024

Consistency Training with Learnable Data Augmentation for Graph Anomaly Detection with Limited Supervision

ICLR 2024spotlight

Graph Anomaly Detection (GAD) has surfaced as a significant field of research, predominantly due to its substantial influence in production environments. Although existing approaches for node anomaly detection have shown effectiveness, they have yet to fully address two major challenges: operating i…

2024

Exploring Collaboration Mechanisms for LLM Agents: A Social Psychology View

ACL 2024long

As Natural Language Processing (NLP) systems are increasingly employed in intricate social environments, a pressing query emerges: *Can these NLP systems mirror human-esque collaborative intelligence, in a multi-agent society consisting of multiple large language models (LLMs)?* This paper probes th…

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…

2024

Harnessing Explanations: LLM-to-LM Interpreter for Enhanced Text-Attributed Graph Representation Learning

ICLR 2024poster

Representation learning on text-attributed graphs (TAGs) has become a critical research problem in recent years. A typical example of a TAG is a paper citation graph, where the text of each paper serves as node attributes. Initial graph neural network (GNN) pipelines handled these text attributes by…

2024

Partitioning Message Passing for Graph Fraud Detection

ICLR 2024poster

Label imbalance and homophily-heterophily mixture are the fundamental problems encountered when applying Graph Neural Networks (GNNs) to Graph Fraud Detection (GFD) tasks. Existing GNN-based GFD models are designed to augment graph structure to accommodate the inductive bias of GNNs towards homophil…

Cited by 30SourcePDFScholar
2024

PoetryDiffusion: Towards Joint Semantic and Metrical Manipulation in Poetry Generation

AAAI 2024technical

Controllable text generation is a challenging and meaningful field in natural language generation (NLG). Especially, poetry generation is a typical one with well-defined and strict conditions for text generation which is an ideal playground for the assessment of current methodologies. While prior wo…

2024

Scalable and Effective Implicit Graph Neural Networks on Large Graphs

ICLR 2024poster

Graph Neural Networks (GNNs) have become the de facto standard for modeling graph-structured data in various applications. Among them, implicit GNNs have shown a superior ability to effectively capture long-range dependencies in underlying graphs. However, implicit GNNs tend to be computationally ex…

Cited by 10SourcePDFScholar
2024

Truth Table Net: Scalable, Compact & Verifiable Neural Networks with a Dual Convolutional Small Boolean Circuit Networks Form

IJCAI 2024poster

We introduce "Truth Table net"' (TTnet), a novel Deep Neural Network (DNN) architecture designed to provide excellent scalability/compactness trade-offs among DNNs, allowing in turn to tackle the DNN challenge of fast formal verification. TTnet is constructed using Learning Truth Table (LTT) filters…

Cited by 0SourcePDFScholar
2024

Uncertainty of Thoughts: Uncertainty-Aware Planning Enhances Information Seeking in LLMs

NeurIPS 2024poster

In the face of uncertainty, the ability to *seek information* is of fundamental importance. In many practical applications, such as medical diagnosis and troubleshooting, the information needed to solve the task is not initially given, and has to be actively sought by asking follow-up questions (for…

Cited by 4SourcePDFScholar
2023

A Generalization of ViT/MLP-Mixer to Graphs

ICML 2023poster

Graph Neural Networks (GNNs) have shown great potential in the field of graph representation learning. Standard GNNs define a local message-passing mechanism which propagates information over the whole graph domain by stacking multiple layers. This paradigm suffers from two major limitations, over-s…

2023

Expanding Small-Scale Datasets with Guided Imagination

NeurIPS 2023poster

The power of DNNs relies heavily on the quantity and quality of training data. However, collecting and annotating data on a large scale is often expensive and time-consuming. To address this issue, we explore a new task, termed dataset expansion, aimed at expanding a ready-to-use small dataset by au…

2023

GraphCleaner: Detecting Mislabelled Samples in Popular Graph Learning Benchmarks

ICML 2023poster

Label errors have been found to be prevalent in popular text, vision, and audio datasets, which heavily influence the safe development and evaluation of machine learning algorithms. Despite increasing efforts towards improving the quality of generic data types, such as images and texts, the problem…

2023

Great Models Think Alike: Improving Model Reliability via Inter-Model Latent Agreement

ICML 2023poster

Reliable application of machine learning is of primary importance to the practical deployment of deep learning methods. A fundamental challenge is that models are often unreliable due to overconfidence. In this paper, we estimate a model's reliability by measuring the agreement between its latent sp…

Cited by 5SourcePDFScholar
2023

LargeST: A Benchmark Dataset for Large-Scale Traffic Forecasting

NeurIPS 2023poster

Road traffic forecasting plays a critical role in smart city initiatives and has experienced significant advancements thanks to the power of deep learning in capturing non-linear patterns of traffic data. However, the promising results achieved on current public datasets may not be applicable to pra…

2023

Probabilistic Knowledge Distillation of Face Ensembles

CVPR 2023poster

Mean ensemble (i.e. averaging predictions from multiple models) is a commonly-used technique in machine learning that improves the performance of each individual model. We formalize it as feature alignment for ensemble in open-set face recognition and generalize it into Bayesian Ensemble Averaging (…

2023

Proximity-Informed Calibration for Deep Neural Networks

NeurIPS 2023spotlight

Confidence calibration is central to providing accurate and interpretable uncertainty estimates, especially under safety-critical scenarios. However, we find that existing calibration algorithms often overlook the issue of proximity bias, a phenomenon where models tend to be more overconfident in lo…

2023

Reachability-Aware Laplacian Representation in Reinforcement Learning

ICML 2023poster

In Reinforcement Learning (RL), Laplacian Representation (LapRep) is a task-agnostic state representation that encodes the geometry of the environment. A desirable property of LapRep stated in prior works is that the Euclidean distance in the LapRep space roughly reflects the reachability between st…

Cited by 4SourcePDFScholar
2023

SPEECH: Structured Prediction with Energy-Based Event-Centric Hyperspheres

ACL 2023long

Event-centric structured prediction involves predicting structured outputs of events. In most NLP cases, event structures are complex with manifold dependency, and it is challenging to effectively represent these complicated structured events. To address these issues, we propose Structured Predictio…

2023

Towards Better Graph Representation Learning with Parameterized Decomposition & Filtering

ICML 2023poster

Proposing an effective and flexible matrix to represent a graph is a fundamental challenge that has been explored from multiple perspectives, e.g., filtering in Graph Fourier Transforms. In this work, we develop a novel and general framework which unifies many existing GNN models from the view of pa…

2022

"Trust, but Verify: Using Self-Supervised Probing to Improve Trustworthiness"

ECCV 2022poster

"Trustworthy machine learning is of primary importance to the practical deployment of deep learning models. While state-of-the-art models achieve astonishingly good performance in terms of accuracy, recent literature reveals that their predictive confidence scores unfortunately cannot be trusted: e.…

2022

CADET: Calibrated Anomaly Detection for Mitigating Hardness Bias

IJCAI 2022poster

The detection of anomalous samples in large, high-dimensional datasets is a challenging task with numerous practical applications. Recently, state-of-the-art performance is achieved with deep learning methods: for example, using the reconstruction error from an autoencoder as anomaly scores. However…

2022

Dangling-Aware Entity Alignment with Mixed High-Order Proximities

NAACL 2022findings

We study dangling-aware entity alignment in knowledge graphs (KGs), which is an underexplored but important problem. As different KGs are naturally constructed by different sets of entities, a KG commonly contains some dangling entities that cannot find counterparts in other KGs. Therefore, dangling…

Cited by 10SourcePDFScholar
2022

GraphCache: Message Passing as Caching for Sentence-Level Relation Extraction

NAACL 2022findings

Entity types and textual context are essential properties for sentence-level relation extraction (RE). Existing work only encodes these properties within individual instances, which limits the performance of RE given the insufficient features in a single sentence. In contrast, we model these propert…

Cited by 8SourcePDFScholar
2022

LUNAR: Unifying Local Outlier Detection Methods via Graph Neural Networks

AAAI 2022technical

Many well-established anomaly detection methods use the distance of a sample to those in its local neighbourhood: so-called `local outlier methods', such as LOF and DBSCAN. They are popular for their simple principles and strong performance on unstructured, feature-based data that is commonplace in…

2022

MGNNI: Multiscale Graph Neural Networks with Implicit Layers

NeurIPS 2022accept

Recently, implicit graph neural networks (GNNs) have been proposed to capture long-range dependencies in underlying graphs. In this paper, we introduce and justify two weaknesses of implicit GNNs: the constrained expressiveness due to their limited effective range for capturing long-range dependenci…

2022

Self-Supervised Aggregation of Diverse Experts for Test-Agnostic Long-Tailed Recognition

NeurIPS 2022accept

Existing long-tailed recognition methods, aiming to train class-balanced models from long-tailed data, generally assume the models would be evaluated on the uniform test class distribution. However, practical test class distributions often violate this assumption (e.g., being either long-tailed or e…

2022

Should We Rely on Entity Mentions for Relation Extraction? Debiasing Relation Extraction with Counterfactual Analysis

NAACL 2022long

Recent literature focuses on utilizing the entity information in the sentence-level relation extraction (RE), but this risks leaking superficial and spurious clues of relations. As a result, RE still suffers from unintended entity bias, i.e., the spurious correlation between entity mentions (names)…

2022

The Geometry of Robust Value Functions

ICML 2022spotlight

The space of value functions is a fundamental concept in reinforcement learning. Characterizing its geometric properties may provide insights for optimization and representation. Existing works mainly focus on the value space for Markov Decision Processes (MDPs). In this paper, we study the geometry…

Cited by 8SourcePDFScholar
2021

Adaptive Data Augmentation on Temporal Graphs

NeurIPS 2021poster

Temporal Graph Networks (TGNs) are powerful on modeling temporal graph data based on their increased complexity. Higher complexity carries with it a higher risk of overfitting, which makes TGNs capture random noise instead of essential semantic information. To address this issue, our idea is to tran…

Cited by 66SourcePDFScholar
2021

EIGNN: Efficient Infinite-Depth Graph Neural Networks

NeurIPS 2021poster

Graph neural networks (GNNs) are widely used for modelling graph-structured data in numerous applications. However, with their inherently finite aggregation layers, existing GNN models may not be able to effectively capture long-range dependencies in the underlying graphs. Motivated by this limitati…

2021

SSMF: Shifting Seasonal Matrix Factorization

NeurIPS 2021poster

Given taxi-ride counts information between departure and destination locations, how can we forecast their future demands? In general, given a data stream of events with seasonal patterns that innovate over time, how can we effectively and efficiently forecast future events? In this paper, we propose…

2021

Spherical Confidence Learning for Face Recognition

CVPR 2021poster

An emerging line of research has found that spherical spaces better match the underlying geometry of facial images, as evidenced by the state-of-the-art facial recognition methods which benefit empirically from spherical representations. Yet, these approaches rely on deterministic embeddings and hen…

Cited by 92PDFcodeScholar
2021

Towards Better Laplacian Representation in Reinforcement Learning with Generalized Graph Drawing

ICML 2021spotlight

The Laplacian representation recently gains increasing attention for reinforcement learning as it provides succinct and informative representation for states, by taking the eigenvectors of the Laplacian matrix of the state-transition graph as state embeddings. Such representation captures the geomet…

2021

Unleashing the Power of Contrastive Self-Supervised Visual Models via Contrast-Regularized Fine-Tuning

NeurIPS 2021poster

Contrastive self-supervised learning (CSL) has attracted increasing attention for model pre-training via unlabeled data. The resulted CSL models provide instance-discriminative visual features that are uniformly scattered in the feature space. During deployment, the common practice is to directly f…

2020

Robustness of Autoencoders for Anomaly Detection Under Adversarial Impact

IJCAI 2020poster

Detecting anomalies is an important task in a wide variety of applications and domains. Deep learning methods have achieved state-of-the-art performance in anomaly detection in recent years; unsupervised methods being particularly popular. However, deep learning methods can be fragile to small pertu…

Cited by 0SourcePDFScholar