This repository hosts a fine-tuned version of the DeBERTa-v3-base model for Zero-Shot Text Classification. With this model, you can classify text into predefined categories without needing to retrain the model for specific tasks. It is ideal for tasks where labeled data is scarce or for rapid prototyping of text classification solutions.

Model Overview

Base Model: DeBERTa-v3-base
Architecture: DebertaV2ForSequenceClassification
Language: English (en)
Data Type: float16 (SafeTensor format for efficiency)

This model leverages the capabilities of DeBERTa-v3-base, fine-tuned on datasets like multi_nli, facebook/anli, and fever to enhance its zero-shot classification performance.

Features

Zero-Shot Classification: Directly classify text into any set of user-defined labels without additional training.
Multi-Label Support: Handle tasks with overlapping categories or multiple applicable labels (set multi_label=True).
Pretrained Efficiency: Optimized for inference using mixed-precision (float16) with SafeTensors.

Example Usage

This model is designed to integrate seamlessly with Hugging Face's transformers library. Here's a simple example:

python

from transformers import pipeline

# Load the Zero-Shot Classification pipeline
classifier = pipeline("zero-shot-classification", model="syedkhalid076/DeBERTa-Zero-Shot-Classification")

# Input sequence and candidate labels
sequence_to_classify = "Last week I upgraded my iOS version and ever since then your app is crashing."
candidate_labels = ["mobile", "website", "billing", "account access", "app crash"]

# Perform classification
output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
print(output)

Applications

I trained this model for UX research purposes, but it can be used for any of the following tasks:

Customer Feedback Analysis: Categorize user reviews or feedback.
Intent Detection: Identify user intents in conversational AI systems.
Content Classification: Classify articles, social media posts, or documents.
Error Detection: Detect error reports in logs or feedback.

Training Data

The model was fine-tuned on the following datasets:

MultiNLI: Multi-genre natural language inference corpus.
ANLI: Adversarial NLI dataset for robust entailment modeling.
FEVER: Fact Extraction and Verification dataset.

These datasets help the model generalize across a wide range of zero-shot classification tasks.

Performance

This model demonstrates strong performance across various zero-shot classification benchmarks, effectively distinguishing between user-defined categories in diverse text inputs.

Limitations

Language Support: Currently supports English (en) only.
Context Length: Performance may degrade with extremely long text inputs. Consider truncating inputs to the model's max token length.

License

This model is licensed under the MIT License. You are free to use, modify, and distribute it with appropriate attribution.

Citation

If you use this model in your work, please cite this repository:

bibtex

@misc{syedkhalid076_deberta_zeroshoot,
  author = {Syed Khalid Hussain},
  title = {DeBERTa Zero-Shot Classification},
  year = {2024},
  url = {https://huggingface.co/syedkhalid076/DeBERTa-Zero-Shot-Classification}
}

Acknowledgements

This model was fine-tuned using Hugging Face Transformers and hosted on the Hugging Face Model Hub. Special thanks to the creators of DeBERTa-v3 and the contributing datasets.