MyPoliBERT

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

MyPoliBERT

Model Overview

MyPoliBERT is a fine-tuned version of bert-base-uncased designed to classify political texts in Malaysia. This model performs multi-label and multi-class classification for 12 political topics (Democracy, Economy, Race, Leadership, Development, Corruption, Instability, Safety, Administration, Education, Religion, Environment) with four sentiment classes (Unknown: 0, Negative: 1, Neutral: 2, Positive: 3). The training data comprises diverse sources, including Malaysian news articles, Reddit posts, and Instagram content.

Intended Uses and Limitations

• Intended Uses:
  This model is intended for analyzing political texts (such as news articles and social media posts) in a Malaysian context. It identifies which political topics are mentioned and predicts the sentiment (polarity) for each topic.

• Limitations:

  1. Language and Context Dependency: The model is designed for English texts related to Malaysian politics. Performance on texts in other languages or from different cultural and political contexts may not be reliable.
  2. Offline Learning: The model is trained on static data and does not incorporate trends or events that occurred after training. Periodic retraining or updates may be necessary to reflect the latest political developments.
  3. Interpretability: As with most deep learning models, the predictions are not inherently interpretable. Human review and explainability techniques are recommended for validating results.

Dataset

• Data Sources:

  • tnwei/ms-newspapers dataset
  • Malaysian political posts from Reddit
  • Malaysian political posts from Instagram

  These sources were combined into a single dataset containing approximately 24,260 records. 80% of the dataset was used for training, and 20% was reserved for validation.

• Task:
  The model performs multi-task learning, simultaneously predicting 12 topics and their respective sentiment classes.

Model Architecture

• Base Model: bert-base-uncased
• Output Layer: The model generates logits for 12 topics, each with four sentiment classes (Unknown, Negative, Neutral, Positive).

Training Procedure

• Hyperparameters:

  • learning_rate: 5e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 16
  • optimizer: Adam(betas=(0.9,0.999), epsilon=1e-08)
  • lr_scheduler_type: linear
  • num_epochs: 8
  • mixed_precision_training: Native AMP (fp16)

• Training Configuration (TrainingArguments):

  • evaluation_strategy: "epoch"
  • save_strategy: "epoch"
  • load_best_model_at_end: True
  • metric_for_best_model: "overall_f1"
  • greater_is_better: True

• Custom Trainer:
  The compute_loss method calculates the cross-entropy loss for each label and averages the losses across all labels.

Evaluation and Performance

It achieves the following results on the evaluation set:

• Loss: 0.2776
• Democracy F1: 0.9521
• Democracy Accuracy: 0.9530
• Economy F1: 0.9330
• Economy Accuracy: 0.9338
• Race F1: 0.9499
• Race Accuracy: 0.9503
• Leadership F1: 0.8086
• Leadership Accuracy: 0.8065
• Development F1: 0.8926
• Development Accuracy: 0.8963
• Corruption F1: 0.9513
• Corruption Accuracy: 0.9524
• Instability F1: 0.9211
• Instability Accuracy: 0.9250
• Safety F1: 0.9263
• Safety Accuracy: 0.9266
• Administration F1: 0.8900
• Administration Accuracy: 0.8963
• Education F1: 0.9584
• Education Accuracy: 0.9596
• Religion F1: 0.9408
• Religion Accuracy: 0.9411
• Environment F1: 0.9828
• Environment Accuracy: 0.9829
• Overall F1: 0.9256
• Overall Accuracy: 0.9270

These results indicate that the model demonstrates robust performance across most topics, with high accuracy and F1 scores. However, the performance for some topics, such as Leadership, is relatively lower, suggesting room for improvement through additional data or model refinement.

Training Procedure

• Hyperparameters:

  • learning_rate: 5e-05
  • train_batch_size: 8
  • eval_batch_size: 8
  • seed: 42
  • gradient_accumulation_steps: 2
  • total_train_batch_size: 16
  • optimizer: Adam(betas=(0.9,0.999), epsilon=1e-08)
  • lr_scheduler_type: linear
  • num_epochs: 8
  • mixed_precision_training: Native AMP (fp16)

• Training Configuration (TrainingArguments):

  • evaluation_strategy: "epoch"
  • save_strategy: "epoch"
  • load_best_model_at_end: True
  • metric_for_best_model: "overall_f1"
  • greater_is_better: True

• Custom Trainer:
  The compute_loss method calculates the cross-entropy loss for each label and averages the losses across all labels.

The following hyperparameters were used during training:

• learning_rate: 5e-05
• train_batch_size: 8
• eval_batch_size: 8
• seed: 42
• gradient_accumulation_steps: 2
• total_train_batch_size: 16
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• num_epochs: 8
• mixed_precision_training: Native AMP

Training results

Future Improvements

• Incorporating additional data and fine-tuning to adapt to new languages and emerging political trends.
• Enhancing data balance for underperforming topics, particularly Leadership.
• Introducing explainability techniques (e.g., XAI) to improve model interpretability.

License and Usage Notes

• The model's predictions should be used as a reference and interpreted within the context of the data sources and Malaysian political environment.
• Users are encouraged to validate outputs with human review and consider the limitations of the model.
• Regular updates and retraining are recommended to ensure the model remains relevant and accurate.

Framework versions

• Transformers 4.18.0
• Pytorch 2.5.0+cu121
• Datasets 3.1.0
• Tokenizers 0.12.1