Проектирование архитектуры ML-пайплайна

Проектирование архитектуры ML-пайплайна

ML-пайплайн — воспроизводимая последовательность шагов от сырых данных до работающей модели в production. Без формальной архитектуры ML-проект превращается в ноутбуки на ноутбуке data scientist'а, которые «работают у меня». Правильная архитектура делает модели воспроизводимыми, версионируемыми и масштабируемыми.

Компоненты Production ML-пайплайна

Каждый production ML-пайплайн состоит из нескольких независимо развёртываемых этапов:

Raw Data → [Data Ingestion] → [Feature Engineering] → [Training] → [Evaluation] → [Registry] → [Serving]
                                        ↑                                                           ↓
                              [Feature Store]                                          [Monitoring]

Data Ingestion Layer: получение данных из источников (S3, базы данных, стриминг). Идемпотентность — перезапуск не создаёт дубликаты. Партиционирование по дате для инкрементальных пересчётов.

Feature Engineering Layer: трансформации воспроизводимы и протестированы. Разделение на online features (вычисляются в реальном времени) и batch features (предвычислены). Feature store — единый источник правды.

Training Layer: гиперпараметрический поиск, кросс-валидация, логирование экспериментов. Checkpoint'ы для длительных обучений.

Evaluation Layer: автоматическое сравнение с baseline и champion моделью. Отказ в регистрации при деградации метрик.

Model Registry: версионирование моделей, метаданные, статусы (staging/production/archived).

Serving Layer: inference-сервис с мониторингом latency, throughput, drift.

Выбор оркестратора

Типовая реализация с ZenML

from zenml import step, pipeline
from zenml.steps import Output

@step
def data_ingestion(
    source_path: str,
    start_date: str,
    end_date: str
) -> Output(data=pd.DataFrame):
    """Загрузка данных за период. Идемпотентно."""
    return load_from_s3(source_path, start_date, end_date)

@step
def feature_engineering(
    data: pd.DataFrame
) -> Output(features=pd.DataFrame, feature_metadata=dict):
    """Трансформации. Те же трансформации применяются при инференсе."""
    transformer = FeatureTransformer()
    features = transformer.fit_transform(data)
    # Сохраняем артефакт трансформера для serving
    return features, transformer.get_metadata()

@step
def model_training(
    features: pd.DataFrame,
    hyperparams: dict
) -> Output(model=Any, metrics=dict):
    model = XGBClassifier(**hyperparams)
    X, y = split_features_target(features)
    model.fit(X, y)
    metrics = evaluate_model(model, X, y)
    return model, metrics

@step
def model_evaluation(
    model: Any,
    metrics: dict,
    baseline_metrics: dict
) -> Output(passed=bool):
    """Не пропускаем модель в registry если хуже baseline."""
    return metrics["f1"] > baseline_metrics["f1"] * 0.99

@pipeline
def training_pipeline(source: str, start_date: str, end_date: str, hyperparams: dict):
    data = data_ingestion(source, start_date, end_date)
    features, feature_metadata = feature_engineering(data)
    model, metrics = model_training(features, hyperparams)
    passed = model_evaluation(model, metrics, load_baseline_metrics())
    if passed:
        register_model(model, metrics, feature_metadata)

Feature Store — предотвращение Training-Serving Skew

Training-serving skew — главный источник деградации ML-моделей в production. Причина: фичи считаются по-разному при обучении и инференсе. Feature store решает это: одна кодовая база определяет фичу, она используется и при обучении, и при сервинге.

from feast import FeatureStore, Entity, FeatureView, Field
from feast.types import Float64, Int64

# Определение один раз
customer_stats = FeatureView(
    name="customer_stats",
    entities=["customer_id"],
    ttl=timedelta(days=1),
    schema=[
        Field(name="total_purchases_7d", dtype=Float64),
        Field(name="avg_order_value", dtype=Float64),
        Field(name="days_since_last_purchase", dtype=Int64),
    ],
    source=customer_stats_batch_source,
)

# При обучении
training_df = store.get_historical_features(
    entity_df=entity_df,
    features=["customer_stats:total_purchases_7d", "customer_stats:avg_order_value"]
).to_df()

# При инференсе — те же фичи, те же вычисления
online_features = store.get_online_features(
    features=["customer_stats:total_purchases_7d"],
    entity_rows=[{"customer_id": "12345"}]
).to_dict()

Data Versioning и воспроизводимость

Каждый запуск пайплайна должен быть воспроизводим через 6 месяцев. Требования:

• Версионирование данных: DVC или Delta Lake (time travel)
• Версионирование кода: git commit hash в метаданных модели
• Версионирование окружения: Docker image digest
• Версионирование конфигурации: параметры в YAML, не в коде

# experiment_config.yaml — все параметры в одном месте
data:
  source: s3://bucket/data/
  start_date: "2024-01-01"
  end_date: "2024-12-31"
  version: "v2.3"

features:
  categorical_encoding: "ordinal"
  numerical_scaling: "standard"
  handle_missing: "median"

model:
  type: "lgbm"
  n_estimators: 500
  learning_rate: 0.05
  max_depth: 6
  num_leaves: 31

CI/CD для ML-пайплайнов

# .github/workflows/ml-pipeline.yml
on:
  push:
    paths:
      - 'pipelines/**'
      - 'features/**'

jobs:
  test-and-train:
    steps:
      - name: Unit tests для feature engineering
        run: pytest tests/features/ -v

      - name: Integration test на subset данных
        run: python run_pipeline.py --mode=test --data-fraction=0.01

      - name: Full training run
        if: github.ref == 'refs/heads/main'
        run: python run_pipeline.py --mode=full

      - name: Model evaluation gate
        run: python evaluate_model.py --fail-on-degradation

Мониторинг пайплайна

Метрики, которые нужно отслеживать: время выполнения каждого шага, объём обработанных данных, distribution входных данных (data drift), метрики модели на validation set. Алерты при: падении шага, аномальном изменении метрик данных, деградации model metrics ниже threshold.

Сроки проектирования

Неделя 1–2: Аудит существующего ML-стека, выбор инструментов, проектирование архитектуры

Неделя 3–4: Реализация базового пайплайна для одной модели

Месяц 2: Feature store, model registry, evaluation gate

Месяц 3: CI/CD, мониторинг, документация. Перенос второй модели на новую архитектуру