用LoRA优化GPT-2,在多任务上表现最佳,比新模型更有效。
GPT Meets Graphs and KAN Splines: Testing Novel Frameworks on Multitask Fine-Tuned GPT-2 with LoRA
- 用LoRA微调GPT-2,结合正则化提升性能。
- 最优模型在SST测试集达55.249%准确率,CFIMDB达99.18%。
- 新引入的KAN和图结构反而不如简单优化的LoRA,适合追求高效适配者。
我们探索将可学习且可解释的模块——特别是科尔莫戈罗夫-阿诺尔德网络(KAN)和基于图的表示——融入预训练的GPT-2模型,以提升多任务学习的准确性。受近期在思维链(CoT)模型中使用KAN和图注意力(GAT)架构的推动及其与简单多层感知机(MLP)对比的争议启发,我们首先通过低秩适应(LoRA)增强标准自注意力变压器,优化超参数并引入L2正则化。该方法取得显著改进。为进一步提升可解释性和丰富表示能力,我们开发了两种变体:图式LoRA和混合式KAN LoRA(可学习GPT)。然而系统评估显示,这两种变体均未超越优化后的LoRA增强型变压器。后者在SST测试集上达到55.249%准确率,CFIMDB开发集上为99.18%,释义检测测试准确率为89.9%。在十四行诗生成任务中,获得42.097的CHRF得分。结果表明,通过LoRA实现的高效参数适配仍是本任务中最有效的策略,涵盖情感分析、释义检测和诗歌生成。
原文摘要 · Abstract (English)
We explore the potential of integrating learnable and interpretable modules--specifically Kolmogorov-Arnold Networks (KAN) and graph-based representations--within a pre-trained GPT-2 model to enhance multi-task learning accuracy. Motivated by the recent surge in using KAN and graph attention (GAT) architectures in chain-of-thought (CoT) models and debates over their benefits compared to simpler architectures like MLPs, we begin by enhancing a standard self-attention transformer using Low-Rank Adaptation (LoRA), fine-tuning hyperparameters, and incorporating L2 regularization. This approach yields significant improvements. To further boost interpretability and richer representations, we develop two variants that attempt to improve the standard KAN and GAT: Graph LoRA and Hybrid-KAN LoRA (Learnable GPT). However, systematic evaluations reveal that neither variant outperforms the optimized LoRA-enhanced transformer, which achieves 55.249% accuracy on the SST test set, 99.18% on the CFIMDB dev set, and 89.9% paraphrase detection test accuracy. On sonnet generation, we get a CHRF score of 42.097. These findings highlight that efficient parameter adaptation via LoRA remains the most effective strategy for our tasks: sentiment analysis, paraphrase detection, and sonnet generation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。