Exploration of Plan-Guided Summarization for Narrative Texts: the Case of Small Language Models

🚀 Installation

Instructions for plan_guided_summ python environment:

conda create -n plan_guided_summ python=3.10

Install following packages:

pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu121
pip install transformers==4.41.2
pip install flash_attn==2.5.8
pip install accelerate
pip install datasets
pip install deepspeed
pip install peft
pip install evaluate
pip install rouge-score
pip install absl-py
pip install spacy==3.7.4

🔧 Usage

See shell scripts in the scripts directory for commands to run different experiments.

🤝 Citation

Please consider citing the paper if you use these sources.

@misc{grenander2025explorationplanguidedsummarizationnarrative,
      title={Exploration of Plan-Guided Summarization for Narrative Texts: the Case of Small Language Models}, 
      author={Matt Grenander and Siddharth Varia and Paula Czarnowska and Yogarshi Vyas and Kishaloy Halder and Bonan Min},
      year={2025},
      eprint={2504.09071},
      archivePrefix={arXiv},
      primaryClass={cs.CL},
      url={https://arxiv.org/abs/2504.09071}, 
}