Machine learning models/Production/Swedish Wikipedia article quality
Model card | |
---|---|
This page is an on-wiki machine learning model card. | |
Model Information Hub | |
Model creator(s) | Aaron Halfaker (User:EpochFail) and Amir Sarabadani |
Model owner(s) | WMF Machine Learning Team (ml@wikimediafoundation.org) |
Model interface | Ores homepage |
Code | ORES Github, ORES training data, and ORES model binaries |
Uses PII | No |
In production? | Yes |
Which projects? | Swedish Wikipedia |
This model uses data about a revision to predict the likelihood that the article is of a certain quality level. | |
Motivation
[edit]This model card describes a model for predicting the quality of Wikipedia articles. It uses structural features extracted from the article to label Wikipedia articles with a probability score for each article quality class.
Wikipedia articles range in quality from rich, well-illustrated, fully-referenced articles that fully cover their topic and are easy to read to single sentence stubs that define the topic of the article but do not offer much more information. It is very useful to be able to reliably distinguish between these extremes and the various stages of quality along this spectrum. Wikipedia editors have developed rich rubrics for how to evaluate the quality of Wikipedia articles and are constantly assessing article quality to assist in coordinating work on the wikis. Editors use these quality scores to evaluate and prioritize their work. Researchers use these quality scores to understand content dynamics. Developers use these quality scores as filters when building recommender systems or other tools.
Wikipedia is always changing, which makes it time-consuming (and largely impossible) for editors to keep these quality assessments complete and up-to-date. An automatic quality model can help fill these gaps by evaluating the quality for articles that are unassessed or have changed substantially since they were last assessed. In doing so, it can provide researchers and tool developers with more consistent data and even potentially help editors identify articles that would benefit from a human assessment.
Users and uses
[edit]- high-level analyses of article quality trends
- filtering / ranking articles in tools – e.g. only show low-quality articles in a recommender system
- identifying potential ways to improve articles – e.g. using the lowest-value feature from the model as a recommendation
- projects outside of Swedish Wikipedia
- namespaces outside of 0, disambiguation pages, and redirects
- directly comparing article quality across language editions – the scores are for a given project so e.g., an article that received a 0.5 score on English Wikipedia would get a much higher score if it had been on Simple English Wikipedia instead (because high-quality articles on English Wikipedia generally have more content than high-quality articles on Simple English Wikipedia)
This model is a part of ORES, and generally accessible via API. It is used for high-level analysis of Wikipedia, platform research, and other on-wiki tasks.
Example API call:https://ores.wikimedia.org/v3/scores/svwiki/1234/articlequality
Ethical considerations, caveats, and recommendations
[edit]- The source data for this model is several years old — data drift may skew current outputs relative to the training data.
- The model does not currently take into account the quality of the specific writing, so a long article with many fake words would register as high quality. It does take into account structure though, so a long article would be penalized if it did not have many sections or was poorly referenced.
- Different wikis have different labeling schemes — do not use this model in conjunction with other models to conduct an interwiki analysis.
Model
[edit]Performance
[edit]Test data confusion matrix:
Label | n | ~u | ~b | ~r | ~s |
---|---|---|---|---|---|
u | 498 | 371 | 15 | 106 | 6 |
b | 497 | 87 | 222 | 106 | 82 |
r | 499 | 176 | 23 | 289 | 11 |
s | 466 | 1 | 54 | 15 | 396 |
Test data sample rates:
u | b | r | s | |
---|---|---|---|---|
sample | 0.254 | 0.254 | 0.255 | 0.238 |
population | 0.019 | 0.06 | 0.042 | 0.879 |
Test data performance:
Statistic | u | b | r | s |
---|---|---|---|---|
match_rate | 0.191 | 0.086 | 0.173 | 0.755 |
filter_rate | 0.809 | 0.914 | 0.827 | 0.245 |
recall | 0.745 | 0.447 | 0.579 | 0.85 |
precision | 0.074 | 0.312 | 0.14 | 0.989 |
f1 | 0.135 | 0.367 | 0.226 | 0.914 |
accuracy | 0.818 | 0.908 | 0.833 | 0.86 |
fpr | 0.181 | 0.063 | 0.155 | 0.066 |
roc_auc | 0.874 | 0.761 | 0.826 | 0.959 |
pr_auc | 0.105 | 0.404 | 0.174 | 0.992 |
Implementation
[edit]{
"type": "GradientBoosting",
"params": {
"loss": "deviance",
"init": null,
"ccp_alpha": 0.0,
"verbose": 0,
"min_impurity_split": null,
"center": true,
"multilabel": false,
"n_iter_no_change": null,
"min_samples_split": 2,
"validation_fraction": 0.1,
"subsample": 1.0,
"min_samples_leaf": 1,
"random_state": null,
"max_features": "log2",
"label_weights": null,
"warm_start": false,
"min_weight_fraction_leaf": 0.0,
"labels": [
"u",
"b",
"r",
"s"
],
"population_rates": null,
"tol": 0.0001,
"criterion": "friedman_mse",
"min_impurity_decrease": 0.0,
"presort": "deprecated",
"n_estimators": 500,
"learning_rate": 0.01,
"max_depth": 3,
"max_leaf_nodes": null,
"scale": true
}
}
{
"title": "Scikit learn-based classifier score with probability",
"properties": {
"probability": {
"description": "A mapping of probabilities onto each of the potential output labels",
"properties": {
"u": {
"type": "number"
},
"s": {
"type": "number"
},
"b": {
"type": "number"
},
"r": {
"type": "number"
}
},
"type": "object"
},
"prediction": {
"description": "The most likely label predicted by the estimator",
"type": "string"
}
},
"type": "object"
}
https://ores.wikimedia.org/v3/scores/svwiki/1234/articlequality
Output:
{
"svwiki": {
"models": {
"articlequality": {
"version": "0.8.0"
}
},
"scores": {
"1234": {
"articlequality": {
"score": {
"prediction": "s",
"probability": {
"b": 0.009658583330794227,
"r": 0.005079205879615088,
"s": 0.9790932720055064,
"u": 0.006168938784084233
}
}
}
}
}
}
}
Data
[edit]Licenses
[edit]- Code: MIT license
- Model: MIT license
Citation
[edit]Cite this model card as:
@misc{
Triedman_Bazira_2023_Swedish_Wikipedia_article_quality,
title={ Swedish Wikipedia article quality model card },
author={ Triedman, Harold and Bazira, Kevin },
year={ 2023 },
url={ https://meta.wikimedia.org/wiki/Machine_learning_models/Production/Swedish_Wikipedia_article_quality }
}
For more information about the implementation details of this model, please refer to: https://phabricator.wikimedia.org/T202202 |