← Search

Kezhi Kong

9 accepted papers

2025

Nemotron-CC: Transforming Common Crawl into a Refined Long-Horizon Pretraining Dataset

ACL 2025long

Recent English Common Crawl datasets like FineWeb-Edu and DCLM achieved significant benchmark gains via aggressive model-based filtering, but at the cost of removing 90% of data. This limits their suitability for long token horizon training, such as 15T tokens for Llama 3.1. In this paper, we show h…

2024

On the Reliability of Watermarks for Large Language Models

ICLR 2024poster

As LLMs become commonplace, machine-generated text has the potential to flood the internet with spam, social media bots, and valueless content. _Watermarking_ is a simple and effective strategy for mitigating such harms by enabling the detection and documentation of LLM-generated text. Yet a crucial…

2024

OpenTab: Advancing Large Language Models as Open-domain Table Reasoners

ICLR 2024poster

Large Language Models (LLMs) trained on large volumes of data excel at various natural language tasks, but they cannot handle tasks requiring knowledge that has not been trained on previously. One solution is to use a retriever that fetches relevant information to expand LLM's knowledge scope. Howev…

2023

GOAT: A Global Transformer on Large-scale Graphs

ICML 2023poster

Graph transformers have been competitive on graph classification tasks, but they fail to outperform Graph Neural Networks (GNNs) on node classification, which is a common task performed on large-scale graphs for industrial applications. Meanwhile, existing GNN architectures are limited in their abil…

Cited by 67SourcePDFScholar
2022

Robust Optimization As Data Augmentation for Large-Scale Graphs

CVPR 2022poster

Data augmentation helps neural networks generalize better by enlarging the training set, but it remains an open question how to effectively augment graph data to enhance the performance of GNNs (Graph Neural Networks). While most existing graph regularizers focus on manipulating graph topological st…

Cited by 122PDFcodeScholar
2021

Data Augmentation for Meta-Learning

ICML 2021spotlight

Conventional image classifiers are trained by randomly sampling mini-batches of images. To achieve state-of-the-art performance, practitioners use sophisticated data augmentation schemes to expand the amount of training data available for sampling. In contrast, meta-learning algorithms sample suppor…

2021

GradInit: Learning to Initialize Neural Networks for Stable and Efficient Training

NeurIPS 2021poster

Innovations in neural architectures have fostered significant breakthroughs in language modeling and computer vision. Unfortunately, novel architectures often result in challenging hyper-parameter choices and training instability if the network parameters are not properly initialized. A number of ar…

2021

SHOT-VAE: Semi-supervised Deep Generative Models With Label-aware ELBO Approximations

AAAI 2021technical

Semi-supervised variational autoencoders (VAEs) have obtained strong results, but have also encountered the challenge that good ELBO values do not always imply accurate inference results.In this paper, we investigate and propose two causes of this problem: (1) The ELBO objective cannot utilize the l…

2021

VQ-GNN: A Universal Framework to Scale up Graph Neural Networks using Vector Quantization

NeurIPS 2021poster

Most state-of-the-art Graph Neural Networks (GNNs) can be defined as a form of graph convolution which can be realized by message passing between direct neighbors or beyond. To scale such GNNs to large graphs, various neighbor-, layer-, or subgraph-sampling techniques are proposed to alleviate the "…