← Search

Xiao Shi Huang

4 accepted papers

2025

Document Summarization with Conformal Importance Guarantees

NeurIPS 2025poster

Automatic summarization systems have advanced rapidly with large language models (LLMs), yet they still lack reliable guarantees on inclusion of critical content in high-stakes domains like healthcare, law, and finance. In this work, we introduce Conformal Importance Summarization, the first framewo…

Cited by 0SourcecodeScholar
2024

Self-supervised Representation Learning from Random Data Projectors

ICLR 2024poster

Self-supervised representation learning (SSRL) has advanced considerably by exploiting the transformation invariance assumption under artificially designed data augmentations. While augmentation-based SSRL algorithms push the boundaries of performance in computer vision and natural language processi…

2022

Improving Non-Autoregressive Translation Models Without Distillation

ICLR 2022poster

Transformer-based autoregressive (AR) machine translation models have achieved significant performance improvements, nearing human-level accuracy on some languages. The AR framework translates one token at a time which can be time consuming, especially for long sequences. To accelerate inference, re…

Cited by 54SourcePDFScholar
2020

Improving Transformer Optimization Through Better Initialization

ICML 2020poster

The Transformer architecture has achieved considerable success recently; the key component of the Transformer is the attention layer that enables the model to focus on important regions within an input sequence. Gradient optimization with attention layers can be notoriously difficult requiring trick…