让AI模仿布莱希特风格对话，生成有戏剧张力的对白。
BrAIcht, a theatrical agent that speaks like Bertolt Brecht's characters
- 用29部布莱希特剧作+907部相似风格德语剧作微调模型。
- 在BLEU和困惑度指标上表现优异，风格还原度高。
- 适合戏剧创作、文学研究者或对德国表现主义感兴趣的读者。
本项目提出BrAIcht，一个能以著名德国剧作家贝托尔特·布莱希特风格进行对话的AI代理。模型基于拥有70亿参数的德语大语言模型LeoLM（Llama2的改良版）进行微调。训练数据包含29部布莱希特原创剧作及907部风格相近的德语剧作，构建更丰富的语料库。受限于内存，采用参数高效微调技术QLoRA。实验结果显示，该模型在生成布莱希特式对话方面表现出色，基于BLEU分数与困惑度评估均达到理想水平。
原文摘要 · Abstract (English)
This project introduces BrAIcht, an AI conversational agent that creates dialogues in the distinctive style of the famous German playwright Bertolt Brecht. BrAIcht is fine-tuned using German LeoLM, a large language model with 7 billion parameters and a modified version of the base Llama2 suitable for German language tasks. For fine-tuning, 29 plays of Bertolt Brecht and 907 of other German plays that are stylistically similar to Bertolt Brecht are used to form a more di-erse dataset. Due to the limited memory capacity, a parameterefficient fine-tuning technique called QLoRA is implemented to train the large language model. The results, based on BLEU score and perplexity, show very promising performance of BrAIcht in generating dialogues in the style of Bertolt Brecht.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品；中文卡片由大模型生成，请以原文为准。