← Search

Yangkun Wang

8 accepted papers

2024

Towards Few-shot Entity Recognition in Document Images: A Graph Neural Network Approach Robust to Image Manipulation

COLING 2024main

Recent advances of incorporating layout information, typically bounding box coordinates, into pre-trained language models have achieved significant performance in entity recognition from document images. Using coordinates can easily model the position of each token, but they are sensitive to manipul…

2023

ToxicChat: Unveiling Hidden Challenges of Toxicity Detection in Real-World User-AI Conversation

EMNLP 2023short findings

Despite remarkable advances that large language models have achieved in chatbots nowadays, maintaining a non-toxic user-AI interactive environment has become increasingly critical nowadays. However, previous efforts in toxicity detection have been mostly based on benchmarks derived from social media…

Cited by 0SourceScholar
2022

Does your graph need a confidence boost? Convergent boosted smoothing on graphs with tabular node features

ICLR 2022spotlight

Many practical modeling tasks require making predictions using tabular data composed of heterogeneous feature types (e.g., text-based, categorical, continuous, etc.). In this setting boosted decision trees and related ensembling techniques generally dominate real-world applications involving iid tr…

2022

Inductive Relation Prediction Using Analogy Subgraph Embeddings

ICLR 2022poster

Prevailing methods for relation prediction in heterogeneous graphs aim at learning latent representations (i.e., embeddings) of observed nodes and relations, and thus are limited to the transductive setting where the relation types must be known during training. Here, we propose ANalogy SubGraphE…

Cited by 7SourcePDFScholar
2022

Learning Enhanced Representation for Tabular Data via Neighborhood Propagation

NeurIPS 2022accept

Prediction over tabular data is an essential and fundamental problem in many important downstream tasks. However, existing methods either take a data instance of the table independently as input or do not fully utilize the multi-row features and labels to directly change and enhance the target data…

2022

Why Propagate Alone? Parallel Use of Labels and Features on Graphs

ICLR 2022poster

One of the challenges of graph-based semi-supervised learning over ordinary supervised learning for classification tasks lies in label utilization. The direct use of ground-truth labels in graphs for training purposes can result in a parametric model learning trivial degenerate solutions (e.g., an…

Cited by 12SourcePDFScholar
2021

Graph Neural Networks Inspired by Classical Iterative Algorithms

ICML 2021oral

Despite the recent success of graph neural networks (GNN), common architectures often exhibit significant limitations, including sensitivity to oversmoothing, long-range dependencies, and spurious edges, e.g., as can occur as a result of graph heterophily or adversarial attacks. To at least partiall…