← Search

Yanqi Shi

2 accepted papers

2024

3D Parallelism for Transformers via Integer Programming

ICASSP 2024accepted

Transformer models, such as BERT, GPT, and ViT, have been applied to a wide range of areas in recent years, due to their efficacy. In order to improve the training efficiency of Transformer models, different distributed training approaches have been proposed, like Megatron-LM [8]. However, when mult…

Cited by 0SourceScholar
2023

GROVE: A Retrieval-augmented Complex Story Generation Framework with A Forest of Evidence

EMNLP 2023long findings

Conditional story generation is significant in human-machine interaction, particularly in producing stories with complex plots. While Large language models (LLMs) perform well on multiple NLP tasks, including story generation, it is challenging to generate stories with both complex and creative plot…

Cited by 0SourceScholar