Saltar al contenido principal

Types of Training Tasks

AI Training supports many different tasks. Each task is optimized for specific types of problems you want to solve.

Text Tasks

Text Classification

What it does: Sorts text into categories you define. Real-world examples:
  • Email spam detection
  • Customer feedback sentiment (happy/unhappy)
  • Support ticket routing
  • Content moderation
  • Language detection
What you need:
  • Text samples
  • Category labels for each sample
  • At least 50 examples per category (more is better)
Example data format:

Token Classification (NER)

What it does: Labels specific words or phrases in text. Real-world examples:
  • Extract names, dates, locations from documents
  • Identify product mentions in reviews
  • Find medical terms in clinical notes
  • Highlight important contract clauses
  • Tag parts of speech
What you need:
  • Text with marked entities
  • BIO format (Beginning, Inside, Outside) labels
  • Hundreds of annotated examples
Example data format:

Sequence to Sequence

What it does: Transforms one text into another. Real-world examples:
  • Language translation
  • Text summarization
  • Question answering
  • Text correction
  • Paraphrasing
What you need:
  • Input text
  • Desired output text
  • Pairs of input-output examples
Example data format:

Text Generation (LLM Fine-tuning)

What it does: Teaches language models new knowledge or behaviors. Real-world examples:
  • Custom chatbots
  • Domain-specific assistants
  • Code generation
  • Creative writing
  • Technical documentation
What you need:
  • Conversation examples or instruction-response pairs
  • Optional: Preference data for RLHF
  • Can work with as few as 100 examples
Example conversation format:

Image Tasks

Image Classification

What it does: Identifies what’s in an image. Real-world examples:
  • Product quality inspection
  • Medical image diagnosis
  • Wildlife identification
  • Document type classification
  • Facial expression recognition
What you need:
  • Images (JPG, PNG)
  • Category label for each image
  • At least 100 images per category
Folder structure:

Object Detection

What it does: Finds and locates multiple objects in images. Real-world examples:
  • Inventory counting
  • Security monitoring
  • Autonomous driving
  • Defect detection
  • People counting
What you need:
  • Images with bounding boxes
  • Labels for each box
  • COCO or YOLO format annotations

Structured Data Tasks

Tabular Classification

What it does: Predicts categories from spreadsheet-like data. Real-world examples:
  • Customer churn prediction
  • Fraud detection
  • Disease diagnosis
  • Credit approval
  • Equipment failure prediction
What you need:
  • CSV with features and labels
  • Numerical and categorical columns
  • Clean, preprocessed data
Example data:

Tabular Regression

What it does: Predicts continuous values from structured data. Real-world examples:
  • House price prediction
  • Sales forecasting
  • Energy consumption estimation
  • Stock price prediction
  • Delivery time estimation
What you need:
  • CSV with features and target values
  • Numerical target column
  • Historical data

Advanced Training Methods

Supervised Fine-Tuning (SFT)

Standard training with examples and correct answers. Use when: You have good quality labeled data.

DPO (Direct Preference Optimization)

Train models using preference comparisons. Use when: You have examples of good vs bad outputs. Data format:

ORPO (Odds Ratio Preference Optimization)

Similar to DPO but more stable training. Use when: DPO training is unstable or overfitting.

Reward Modeling

Train a model to score outputs. Use when: Building a reward model for RLHF.

PPO (Proximal Policy Optimization)

Reinforcement learning from feedback. Use when: You have a reward model and want to optimize against it.

Task Selection Guide

Based on Your Data

Based on Your Goal

Based on Difficulty

Easiest to start:
  1. Text Classification
  2. Image Classification
  3. Tabular Classification
Medium difficulty:
  1. Token Classification
  2. Sequence to Sequence
  3. LLM Fine-tuning (SFT)
Advanced:
  1. DPO/ORPO training
  2. Object Detection
  3. PPO/RLHF

Data Requirements

Minimum Data Needed

Data Quality Matters

Better to have 100 high-quality examples than 1,000 poor ones:
  • Accurate labels
  • Consistent formatting
  • Representative of real-world use
  • Balanced across categories

Multi-Task Training

You can train models for multiple tasks simultaneously:

Benefits

  • Share knowledge between tasks
  • More efficient use of data
  • Single model deployment

Example

Train one model to:
  • Classify sentiment
  • Extract entities
  • Summarize text
All with the same base model.

Task-Specific Settings

Text Tasks

  • Max sequence length: How much text to process
  • Tokenizer: How to split text into tokens
  • Special tokens: Task-specific markers

Image Tasks

  • Image size: Resolution to use
  • Augmentation: Rotation, flip, crop
  • Normalization: Pixel value scaling

Tabular Tasks

  • Feature engineering: Creating new columns
  • Scaling: Normalizing numeric values
  • Encoding: Handling categorical variables

Evaluation Metrics

Different tasks use different metrics:

Combining Tasks

Pipeline Approach

Chain tasks together:
  1. Classification → Route to specialized model
  2. NER → Extract entities → Generate response
  3. Translate → Summarize → Classify sentiment

Multi-Modal Tasks

Combine different data types:
  • Image + Text → Visual QA
  • Audio + Text → Speech recognition
  • Video + Text → Video understanding

Next Steps

Ready to dive deeper?

Model Types

Choose the right model architecture

Quick Start

Train your first model