Selective Training: A Strategy for Fast Backpropagation on Sentence Embeddings

Research output: Contribution to book/anthology/report/proceedingArticle in proceedingsResearchpeer-review

Representation or embedding based machine learning models, such as language models or convolutional neural networks have shown great potential for improved performance. However, for complex models on large datasets training time can be extensive, approaching weeks, which is often infeasible in practice. In this work, we present a method to reduce training time substantially by selecting training instances that provide relevant information for training. Selection is based on the similarity of the learned representations over input instances, thus allowing for learning a non-trivial weighting scheme from multi-dimensional representations. We demonstrate the efficiency and effectivity of our approach in several text classification tasks using recursive neural networks. Our experiments show that by removing approximately one fifth of the training data the objective function converges up to six times faster without sacrificing accuracy.
Original languageEnglish
Title of host publicationAdvances in Knowledge Discovery and Data Mining - 23rd Pacific-Asia Conference, PAKDD 2019, Proceedings : PAKDD '19
EditorsQiang Yang, Min-Ling Zhang, Zhiguo Gong, Sheng-Jun Huang, Zhi-Hua Zhou
Number of pages14
PublisherSpringer
Publication year2019
Pages40-53
ISBN (print)978-3-030-16141-5
ISBN (Electronic)978-3-030-16142-2
DOIs
Publication statusPublished - 2019

    Research areas

  • Selective training, Machine learning, Neural network, Recursive models

See relations at Aarhus University Citationformats

ID: 130411244