Jump to content

Machine learning models/Production/Wikidata damaging edit

From Meta, a Wikimedia project coordination wiki


Model card
This page is an on-wiki machine learning model card.
A diagram of a neural network
A model card is a document about a machine learning model that seeks to answer basic questions about the model.
Model Information Hub
Model creator(s)Aaron Halfaker (User:EpochFail) and Amir Sarabadani
Model owner(s)WMF Machine Learning Team (ml@wikimediafoundation.org)
Model interfaceOres homepage
CodeORES Github, ORES training data, and ORES model binaries
Uses PIINo
In production?Yes
Which projects?Wikidata
This model uses data about a revision to predict the likelihood that the revision is damaging.


Motivation

[edit]

Some goodfaith edits are damaging to an item, and not all damaging edits are in bad faith. This model (together with a goodfaith model) is intended to differentiate between edits that are intentionally harmful (badfaith/vandalism) and edits that are intended to be harmful (good edits/goodfaith damage).

This model helps to prioritize review of potentially damaging edits or vandalism. It provides a prediction on whether or not a given revision is damaging, and provides some probabilities to serve as a measure of its confidence level.

Users and uses

[edit]
Use this model for
  • This model should be used for prioritizing the review and potential reversion of vandalism on Wikidata.
  • This model should be used for detecting damaging contributions by editors on Wikidata.
Don't use this model for
  • This model should not be used as an ultimate arbiter of whether or not an edit ought to be considered damaging.
  • The model should not be used outside of Wikidata.
Current uses
  • Wikidata uses the model as a service for facilitating efficient vandalism triage, edit reviews, or newcomer support.
  • On an individual basis, anyone can submit a properly-formatted API call to ORES for a given revision and get back the result of this model.
Example API call:
https://ores.wikimedia.org/v3/scores/wikidatawiki/1907686315/damaging

Ethical considerations, caveats, and recommendations

[edit]

Wikidata decided to use this model. Over time, the model has been validated through use in the community.

This model is known to give newer editors higher probability of damaging edits.

Internal or external changes that could make this model deprecated or no longer usable are:

  • Data drift means training data for the model is no longer usable.
  • Doesn't meet desired performance metrics in production.
  • Wikidata community decides to not use this model anymore.

Model

[edit]

Performance

[edit]

Test data confusion matrix:

Label n ~True ~False
True 2674 2563 111
False 13461 634 12827

Test data sample rates:

Rate Sample Population
sample 0.166 0.834

Test data performance:

Statistic True False
match_rate 0.198 0.802
filter_rate 0.802 0.198
recall 0.958 0.953
precision 0.802 0.991
f1 0.873 0.972
accuracy 0.954 0.954
fpr 0.047 0.042
roc_auc 0.985 0.985
pr_auc 0.915 0.997

Implementation

[edit]
Model architecture
{
    "type": "GradientBoosting",
    "params": {
        "scale": true,
        "center": true,
        "labels": [
            true,
            false
        ],
        "multilabel": false,
        "population_rates": null,
        "ccp_alpha": 0.0,
        "criterion": "friedman_mse",
        "init": null,
        "learning_rate": 0.01,
        "loss": "deviance",
        "max_depth": 7,
        "max_features": "log2",
        "max_leaf_nodes": null,
        "min_impurity_decrease": 0.0,
        "min_samples_leaf": 1,
        "min_samples_split": 2,
        "min_weight_fraction_leaf": 0.0,
        "n_estimators": 700,
        "n_iter_no_change": null,
        "random_state": null,
        "subsample": 1.0,
        "tol": 0.0001,
        "validation_fraction": 0.1,
        "verbose": 0,
        "warm_start": false
    }
}
Output schema
{
    "title": "Scikit learn-based classifier score with probability",
    "type": "object",
    "properties": {
        "prediction": {
            "description": "The most likely label predicted by the estimator",
            "type": "boolean"
        },
        "probability": {
            "description": "A mapping of probabilities onto each of the potential output labels",
            "type": "object",
            "properties": {
                "true": {
                    "type": "number"
                },
                "false": {
                    "type": "number"
                }
            }
        }
    }
}
Example input and output
Input:
https://ores.wikimedia.org/v3/scores/wikidatawiki/1907686315/damaging

Output:

{
    "wikidatawiki": {
        "models": {
            "damaging": {
                "version": "0.5.0"
            }
        },
        "scores": {
            "1907686315": {
                "damaging": {
                    "score": {
                        "prediction": false,
                        "probability": {
                            "false": 0.9692774805962331,
                            "true": 0.030722519403766885
                        }
                    }
                }
            }
        }
    }
}

Data

[edit]
Data pipeline
Tabular data about edits is collected from the Mediawiki API, preprocessed (via log-transformations, joining with public editor data, etc.), and joined with user-generated goodfaith/damaging labels.
Training data
This model was trained using hand-labeled training data that is several years old.
Test data
The statistics reported here were calculated by selecting a random partition of the training data to hold out from the training process. The model then makes a prediction on that data, which is compared to the underlying ground truth.

Licenses

[edit]

Citation

[edit]

Cite this model card as:

@misc{
  Triedman_Bazira_2023_wikidata_damaging,
  title={ Wikidata damaging model card },
  author={ Triedman, Harold and Bazira, Kevin },
  year={ 2023 },
  url={ https://meta.wikimedia.org/wiki/Machine_learning_models/Production/Wikidata_damaging_edit }
}