đ ALBERT Persian
A Lite BERT for Self-supervised Learning of Language Representations for the Persian Language
ALBERT-Persian is the first attempt on ALBERT for the Persian Language. It can also be called "BERT-Kucholo". ALBERT-Persian is trained based on Google's ALBERT BASE Version 2.0. The training data consists of over 3.9M documents, 73M sentences, and 1.3B words from various writing styles and numerous subjects (e.g., scientific, novels, news), similar to the approach used for ParsBERT.
Please follow the ALBERT-Persian repo for the latest information about previous and current models.
đ Quick Start
⨠Features
- It's the first ALBERT-based model for the Persian language.
- Trained on a large and diverse dataset of Persian texts.
đ Documentation
Persian Text Classification [DigiMag, Persian News]
The task aims to label texts in a supervised manner in both existing datasets DigiMag and Persian News.
DigiMag
A total of 8,515 articles were scraped from Digikala Online Magazine. This dataset includes seven different classes:
- Video Games
- Shopping Guide
- Health Beauty
- Science Technology
- General
- Art Cinema
- Books Literature
| Label |
# |
| Video Games |
1967 |
| Shopping Guide |
125 |
| Health Beauty |
1610 |
| Science Technology |
2772 |
| General |
120 |
| Art Cinema |
1667 |
| Books Literature |
254 |
Download
You can download the dataset from here
Results
The following table summarizes the F1 score obtained by ParsBERT as compared to other models and architectures.
| Dataset |
ALBERT-fa-base-v2 |
ParsBERT-v1 |
mBERT |
| Digikala Magazine |
92.33 |
93.59 |
90.72 |
đ License
This project is licensed under the Apache-2.0 license.
BibTeX entry and citation info
Please cite in publications as the following:
@misc{ALBERTPersian,
author = {Mehrdad Farahani},
title = {ALBERT-Persian: A Lite BERT for Self-supervised Learning of Language Representations for the Persian Language},
year = {2020},
publisher = {GitHub},
journal = {GitHub repository},
howpublished = {\url{https://github.com/m3hrdadfi/albert-persian}},
}
@article{ParsBERT,
title={ParsBERT: Transformer-based Model for Persian Language Understanding},
author={Mehrdad Farahani, Mohammad Gharachorloo, Marzieh Farahani, Mohammad Manthouri},
journal={ArXiv},
year={2020},
volume={abs/2005.12515}
}
Questions?
Post a Github issue on the ALBERT-Persian repo.
%20--%3e%3cdefs%3e%3cstyle%3e%20.st0%20{%20fill:%20%23061b40;%20}%20.st1%20{%20fill:%20%23306af1;%20}%20.st2%20{%20fill:%20%235ce5cf;%20}%20%3c/style%3e%3c/defs%3e%3cg%3e%3cpath%20class='st0'%20d='M55,10.5h9v3h-9v9h12V7.5h-12v3ZM64,19.5h-6v-3h6v3Z'/%3e%3cpolygon%20class='st0'%20points='69%2016.5%2078%2016.5%2078%2019.5%2069%2019.5%2069%2022.5%2081%2022.5%2081%2013.5%2072%2013.5%2072%2010.5%2081%2010.5%2081%207.5%2069%207.5%2069%2016.5'/%3e%3cpolygon%20class='st0'%20points='95%2010.5%2095%207.5%2083%207.5%2083%2022.5%2095%2022.5%2095%2019.5%2086%2019.5%2086%2016.5%2095%2016.5%2095%2013.5%2086%2013.5%2086%2010.5%2095%2010.5'/%3e%3cpath%20class='st0'%20d='M40,1.5v21h11.6l1.4-1.4v-7.6h0c0,0-1.4-1.5-1.4-1.5l1.4-1.4V2.9l-1.4-1.4h-11.6ZM50,19.5h-7v-6h7v6ZM50,10.5h-7v-6h7v6Z'/%3e%3c/g%3e%3cpath%20class='st1'%20d='M23.1,24L14.7,4.8l-4.9,11.2h3.8l-1.8,4H3.3L12.1,0H2C.9,0,0,.9,0,2v20c0,1.1.9,2,2,2h21.1Z'/%3e%3cpath%20class='st2'%20d='M34,0h-16.8l10.6,24h6.2c1.1,0,2-.9,2-2V2C36,.9,35.1,0,34,0ZM32.5,20h-4V4h4v16Z'/%3e%3c/svg%3e)
