Skip to main content
CenXiv.org
此网站处于试运行阶段,支持我们!
我们衷心感谢所有贡献者的支持。
贡献
赞助
cenxiv logo > cs > arXiv:2501.02432v1

帮助 | 高级搜索

计算机科学 > 计算与语言

arXiv:2501.02432v1 (cs)
[提交于 2025年1月5日 ]

标题: Swift跨数据集剪枝:提升自然语言理解中的微调效率

标题: Swift Cross-Dataset Pruning: Enhancing Fine-Tuning Efficiency in Natural Language Understanding

Authors:Binh-Nguyen Nguyen, Yang He
摘要: 数据集剪枝旨在选择数据集的一个子集以实现高效的模型训练。 虽然自然语言处理中的数据效率主要集中在模型预训练期间的语料库内场景,但由于数据集大小、数据分布、类别不平衡和标签空间的差异,跨不同数据集的任务特定微调的高效数据集剪枝仍然具有挑战性。 当前的跨数据集微调剪枝技术通常依赖于计算成本高昂的样本排序过程,通常需要完整的数据集训练或参考模型。 我们通过提出Swift跨数据集剪枝(SCDP)来解决这一问题。 具体而言,我们的方法使用TF-IDF嵌入和几何中位数快速评估样本的重要性。 然后,我们应用与数据集大小适应的剪枝以确保多样性:对于较小的数据集,我们保留远离几何中位数的样本,而对于较大的数据集,我们采用基于距离的分层剪枝。 在六个多样化数据集上的实验结果证明了我们方法的有效性,在各种任务和规模上显著减少了计算资源。 源代码可在以下地址获取:https://github.com/he-y/NLP-Dataset-Pruning
摘要: Dataset pruning aims to select a subset of a dataset for efficient model training. While data efficiency in natural language processing has primarily focused on within-corpus scenarios during model pre-training, efficient dataset pruning for task-specific fine-tuning across diverse datasets remains challenging due to variability in dataset sizes, data distributions, class imbalance and label spaces. Current cross-dataset pruning techniques for fine-tuning often rely on computationally expensive sample ranking processes, typically requiring full dataset training or reference models. We address this gap by proposing Swift Cross-Dataset Pruning (SCDP). Specifically, our approach uses TF-IDF embeddings with geometric median to rapidly evaluate sample importance. We then apply dataset size-adaptive pruning to ensure diversity: for smaller datasets, we retain samples far from the geometric median, while for larger ones, we employ distance-based stratified pruning. Experimental results on six diverse datasets demonstrate the effectiveness of our method, spanning various tasks and scales while significantly reducing computational resources. Source code is available at: https://github.com/he-y/NLP-Dataset-Pruning
评论: 被COLING 2025接收
主题: 计算与语言 (cs.CL)
引用方式: arXiv:2501.02432 [cs.CL]
  (或者 arXiv:2501.02432v1 [cs.CL] 对于此版本)
  https://doi.org/10.48550/arXiv.2501.02432
通过 DataCite 发表的 arXiv DOI

提交历史

来自: Yang He [查看电子邮件]
[v1] 星期日, 2025 年 1 月 5 日 03:52:04 UTC (595 KB)
全文链接:

获取论文:

    查看标题为《》的 PDF
  • 查看中文 PDF
  • 查看 PDF
  • HTML(实验性)
  • TeX 源代码
  • 其他格式
查看许可
当前浏览上下文:
cs.CL
< 上一篇   |   下一篇 >
新的 | 最近的 | 2025-01
切换浏览方式为:
cs

参考文献与引用

  • NASA ADS
  • 谷歌学术搜索
  • 语义学者
a 导出 BibTeX 引用 加载中...

BibTeX 格式的引用

×
数据由提供:

收藏

BibSonomy logo Reddit logo

文献和引用工具

文献资源探索 (什么是资源探索?)
连接的论文 (什么是连接的论文?)
Litmaps (什么是 Litmaps?)
scite 智能引用 (什么是智能引用?)

与本文相关的代码,数据和媒体

alphaXiv (什么是 alphaXiv?)
CatalyzeX 代码查找器 (什么是 CatalyzeX?)
DagsHub (什么是 DagsHub?)
Gotit.pub (什么是 GotitPub?)
Hugging Face (什么是 Huggingface?)
带有代码的论文 (什么是带有代码的论文?)
ScienceCast (什么是 ScienceCast?)

演示

复制 (什么是复制?)
Hugging Face Spaces (什么是 Spaces?)
TXYZ.AI (什么是 TXYZ.AI?)

推荐器和搜索工具

影响之花 (什么是影响之花?)
核心推荐器 (什么是核心?)
IArxiv 推荐器 (什么是 IArxiv?)
  • 作者
  • 地点
  • 机构
  • 主题

arXivLabs:与社区合作伙伴的实验项目

arXivLabs 是一个框架,允许合作伙伴直接在我们的网站上开发和分享新的 arXiv 特性。

与 arXivLabs 合作的个人和组织都接受了我们的价值观,即开放、社区、卓越和用户数据隐私。arXiv 承诺这些价值观,并且只与遵守这些价值观的合作伙伴合作。

有一个为 arXiv 社区增加价值的项目想法吗? 了解更多关于 arXivLabs 的信息.

这篇论文的哪些作者是支持者? | 禁用 MathJax (什么是 MathJax?)
  • 关于
  • 帮助
  • contact arXivClick here to contact arXiv 联系
  • 订阅 arXiv 邮件列表点击这里订阅 订阅
  • 版权
  • 隐私政策
  • 网络无障碍帮助
  • arXiv 运营状态
    通过...获取状态通知 email 或者 slack

京ICP备2025123034号