--- tags: - generated_from_trainer datasets: - xsum metrics: - rouge model-index: - name: t5-small-finetuned_xsum results: - task: name: Sequence-to-sequence Language Modeling type: text2text-generation dataset: name: xsum type: xsum args: default metrics: - name: Rouge1 type: rouge value: 33.1688 --- # t5-small-finetuned_xsum This model is a fine-tuned version of [t5-small](https://huggingface.co/t5-small) on the xsum dataset. It achieves the following results on the evaluation set: - Loss: 2.0881 - Rouge1: 33.1688 - Rouge2: 11.831 - Rougel: 26.796 - Rougelsum: 26.7931 - Gen Len: 18.7957 ## Model description More information needed ## Intended uses & limitations The Extreme Summarization (XSum) dataset is a dataset for evaluation of abstractive single-document summarization systems. The goal is to create a short, one-sentence new summary answering the question “What is the article about?”. The dataset consists of 226,711 news articles accompanied with a one-sentence summary. The articles are collected from BBC articles (2010 to 2017) and cover a wide variety of domains (e.g., News, Politics, Sports, Weather, Business, Technology, Science, Health, Family, Education, Entertainment and Arts). The official random split contains 204,045 (90%), 11,332 (5%) and 11,334 (5) documents in training, validation and test sets, respectively. T5, or Text-to-Text Transfer Transformer, is a Transformer based architecture that uses a text-to-text approach. Every task – including translation, question answering, and classification – is cast as feeding the model text as input and training it to generate some target text. This allows for the use of the same model, loss function, hyperparameters, etc. across our diverse set of tasks. The changes compared to BERT include: - adding a causal decoder to the bidirectional architecture. - replacing the fill-in-the-blank cloze task with a mix of alternative pre-training tasks. ## Training and evaluation data More information needed ## Training procedure ### Training hyperparameters The following hyperparameters were used during training: - learning_rate: 2e-05 - train_batch_size: 16 - eval_batch_size: 16 - seed: 42 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08 - lr_scheduler_type: linear - num_epochs: 50 ### Training results | Training Loss | Epoch | Step | Validation Loss | Rouge1 | Rouge2 | Rougel | Rougelsum | Gen Len | |:-------------:|:-----:|:------:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:| | 2.3789 | 1.0 | 12753 | 2.2274 | 31.3107 | 10.1407 | 25.0522 | 25.0423 | 18.8193 | | 2.3565 | 2.0 | 25506 | 2.2159 | 31.5958 | 10.4022 | 25.3267 | 25.3228 | 18.7992 | | 2.3504 | 3.0 | 38259 | 2.2037 | 31.8838 | 10.5974 | 25.5777 | 25.5786 | 18.7928 | | 2.3345 | 4.0 | 51012 | 2.1956 | 31.8402 | 10.5656 | 25.5027 | 25.4994 | 18.8163 | | 2.3175 | 5.0 | 63765 | 2.1868 | 31.9412 | 10.7187 | 25.6688 | 25.6719 | 18.7902 | | 2.3177 | 6.0 | 76518 | 2.1805 | 31.9831 | 10.7074 | 25.6869 | 25.6863 | 18.8099 | | 2.3027 | 7.0 | 89271 | 2.1734 | 32.0714 | 10.7714 | 25.7193 | 25.7141 | 18.7961 | | 2.289 | 8.0 | 102024 | 2.1667 | 32.1598 | 10.883 | 25.8608 | 25.8605 | 18.8144 | | 2.2875 | 9.0 | 114777 | 2.1622 | 32.0933 | 10.9046 | 25.8399 | 25.8329 | 18.8009 | | 2.2796 | 10.0 | 127530 | 2.1547 | 32.391 | 11.112 | 26.0903 | 26.0931 | 18.7992 | | 2.286 | 11.0 | 140283 | 2.1504 | 32.4479 | 11.1077 | 26.1274 | 26.1267 | 18.7975 | | 2.2542 | 12.0 | 153036 | 2.1464 | 32.4059 | 11.1583 | 26.1111 | 26.1047 | 18.8042 | | 2.2526 | 13.0 | 165789 | 2.1416 | 32.425 | 11.2178 | 26.1854 | 26.1795 | 18.7865 | | 2.2374 | 14.0 | 178542 | 2.1372 | 32.299 | 11.1047 | 26.0495 | 26.0434 | 18.8016 | | 2.2295 | 15.0 | 191295 | 2.1331 | 32.4283 | 11.2233 | 26.135 | 26.128 | 18.8004 | | 2.2213 | 16.0 | 204048 | 2.1306 | 32.4948 | 11.2885 | 26.2607 | 26.2551 | 18.7854 | | 2.1985 | 17.0 | 216801 | 2.1282 | 32.5872 | 11.3243 | 26.31 | 26.3062 | 18.7986 | | 2.1993 | 18.0 | 229554 | 2.1245 | 32.6278 | 11.3196 | 26.3142 | 26.315 | 18.7809 | | 2.2044 | 19.0 | 242307 | 2.1223 | 32.676 | 11.3871 | 26.356 | 26.3426 | 18.8007 | | 2.2035 | 20.0 | 255060 | 2.1188 | 32.8736 | 11.4703 | 26.4901 | 26.4899 | 18.7863 | | 2.1909 | 21.0 | 267813 | 2.1167 | 32.8288 | 11.4666 | 26.4992 | 26.4877 | 18.796 | | 2.1835 | 22.0 | 280566 | 2.1141 | 32.9183 | 11.5267 | 26.5302 | 26.5338 | 18.8034 | | 2.1845 | 23.0 | 293319 | 2.1127 | 32.7907 | 11.444 | 26.4614 | 26.459 | 18.8054 | | 2.1725 | 24.0 | 306072 | 2.1109 | 32.8191 | 11.4973 | 26.5109 | 26.5012 | 18.7818 | | 2.1805 | 25.0 | 318825 | 2.1082 | 32.7333 | 11.4325 | 26.4093 | 26.4028 | 18.7986 | | 2.1661 | 26.0 | 331578 | 2.1063 | 32.8703 | 11.5443 | 26.5105 | 26.5101 | 18.7962 | | 2.1606 | 27.0 | 344331 | 2.1048 | 32.884 | 11.558 | 26.5504 | 26.5465 | 18.7939 | | 2.1508 | 28.0 | 357084 | 2.1032 | 32.9699 | 11.6036 | 26.6348 | 26.6266 | 18.7983 | | 2.1479 | 29.0 | 369837 | 2.1019 | 32.8247 | 11.5812 | 26.5659 | 26.5595 | 18.7992 | | 2.1363 | 30.0 | 382590 | 2.1019 | 32.9982 | 11.6801 | 26.6552 | 26.6497 | 18.797 | | 2.1513 | 31.0 | 395343 | 2.0996 | 32.9903 | 11.6632 | 26.6579 | 26.6521 | 18.7911 | | 2.1389 | 32.0 | 408096 | 2.0981 | 33.0195 | 11.7282 | 26.683 | 26.6757 | 18.7824 | | 2.1421 | 33.0 | 420849 | 2.0968 | 32.9967 | 11.6949 | 26.6734 | 26.662 | 18.796 | | 2.1545 | 34.0 | 433602 | 2.0954 | 33.0943 | 11.7329 | 26.7367 | 26.7295 | 18.7871 | | 2.1459 | 35.0 | 446355 | 2.0949 | 33.1534 | 11.816 | 26.775 | 26.7716 | 18.7914 | | 2.1364 | 36.0 | 459108 | 2.0933 | 33.0686 | 11.7418 | 26.7147 | 26.7066 | 18.7901 | | 2.1194 | 37.0 | 471861 | 2.0928 | 33.1276 | 11.8268 | 26.7684 | 26.7626 | 18.802 | | 2.1292 | 38.0 | 484614 | 2.0925 | 33.0462 | 11.7669 | 26.6798 | 26.6783 | 18.802 | | 2.1317 | 39.0 | 497367 | 2.0913 | 33.1402 | 11.7889 | 26.7822 | 26.7824 | 18.7962 | | 2.1176 | 40.0 | 510120 | 2.0907 | 33.1488 | 11.8001 | 26.7749 | 26.7615 | 18.7992 | | 2.1318 | 41.0 | 522873 | 2.0899 | 33.0963 | 11.8162 | 26.7433 | 26.7325 | 18.7924 | | 2.1052 | 42.0 | 535626 | 2.0899 | 33.0764 | 11.7624 | 26.7294 | 26.7238 | 18.7911 | | 2.1267 | 43.0 | 548379 | 2.0891 | 33.1292 | 11.8029 | 26.7684 | 26.7693 | 18.7885 | | 2.1211 | 44.0 | 561132 | 2.0894 | 33.09 | 11.7676 | 26.7418 | 26.7394 | 18.7853 | | 2.1243 | 45.0 | 573885 | 2.0880 | 33.1449 | 11.7899 | 26.7725 | 26.7634 | 18.7946 | | 2.0947 | 46.0 | 586638 | 2.0885 | 33.1548 | 11.8108 | 26.808 | 26.8003 | 18.7917 | | 2.1246 | 47.0 | 599391 | 2.0881 | 33.148 | 11.8208 | 26.803 | 26.7961 | 18.7913 | | 2.127 | 48.0 | 612144 | 2.0877 | 33.1935 | 11.8399 | 26.8209 | 26.8142 | 18.7925 | | 2.1231 | 49.0 | 624897 | 2.0878 | 33.158 | 11.8159 | 26.7898 | 26.785 | 18.794 | | 2.1296 | 50.0 | 637650 | 2.0881 | 33.1688 | 11.831 | 26.796 | 26.7931 | 18.7957 | ### Framework versions - Transformers 4.12.0.dev0 - Pytorch 1.10.0+cu113 - Datasets 1.14.0 - Tokenizers 0.10.3