Как да подготвим файлови формати за данни за обучение на ИИ и мултимодални LLMs

Последно обновено: 21 май, 2025

TL;DR – Форматът на файла, който изберете, може да намали времето за обучение с 30‑50 %, да намали разходите за съхранение с 1 %–5 % и да предпази вашите мултимодални модели от проблеми с несъгласувани данни. Идеалният избор е потоково‑готов, колоночно ориентиран бинарен контейнер (TFRecord, WebDataset, Arrow/Parquet), който съхранява пред‑токенизиран текст и пред‑кодиран медия в един единствен, версионно‑контролиран шард.

Защо файловият формат е важен за обучението на ИИ

Крайният извод: форматът е първата линия на защита срещу бавно I/O, шумни данни и проблеми със съответствието.

Основни концепции и терминология (Бърз справочник)

Избор на правилния формат

Общ принцип:

• Обучение в мащаб → TFRecord, WebDataset или Arrow/Parquet (те стриймват, компресират и поддържат шардиране).
• Експериментиране → JSONL (човеко‑четим, лесен за редактиране).
• Тежък случай на случайно достъпване (например, генерация, обогатена с извличане) → LMDB.

Стъпка‑по‑стъпка план (От сурови файлове до готови за продукция шарди)

1. Определете единна схема‑източник на истина

   message MultiModalExample {
     bytes image = 1;                // JPEG‑XL or AVIF
     repeated int32 caption = 2;    // token IDs
     bytes audio = 3;                // Opus or FLAC
     map<string, string> meta = 4;  // source_id, timestamp, etc.
   }
   

   Store this .proto (or Arrow schema) alongside the dataset.

2. Събиране и почистване на сурови активи

   • Text: Unicode‑NFKC, премахване на контролни символи, дедупликация.
   • Images: Конвертирайте първо в lossless PNG, след което по избор в lossy JPEG‑XL (качество 85‑90 %).
   • Audio: Пресемплирайте до 16 kHz, 16‑bit PCM; кодирате с Opus (lossy) или FLAC (lossless).

3. Предварителна обработка / Токенизиране
   Използвайте точно токенизатора, който ще подадете на модела (например tiktoken за GPT‑NeoX). Съхранявайте получените int32[] ID‑та на токените директно в записа.

4. Сериализирайте всеки запис
   Изберете бърз бинарен сериализатор: Protocol Buffers, FlatBuffers или Arrow IPC. Целта е един байтов низ за всеки пример, който може да се запише в TFRecord или tarball.

5. Shard & compress

   • Целеви размер на шард: 256 MiB – 1 GiB (оптимален за S3 GET range заявки).
   • Компресирайте с Zstandard (ниво 3‑5) – бързо декомпресиране, добро съотношение.
   • Naming convention: train-00000-of-01000.tfrecord.zst.

6. Generate a manifest

   [
     {
       "shard": "train-00000-of-01000.tfrecord.zst",
       "checksum": "sha256:ab12…",
       "num_examples": 12456,
       "avg_seq_len": 256,
       "git_hash": "d3f9c1e"
     },
     …
   ]
   

   The manifest is the single source of truth for validation, resumable training, and audit.

7. Validate
   Случайно извадете 0.1 % от записите, декодирайте всяко поле и изпълнете проверки за коректност (декодиране на изображение, дължина на токените, продължителност на аудио). Изчислете глобални статистики (покритие на речника, разпределение на резолюцията) и ги съхранете в манифеста.

8. Version & store immutably
   Изпратете шарди + манифест в неизменен bucket (gs://my‑project/datasets/v1/). Маркирайте със семантична версия (v1.0.0) и регистрирайте моментната снимка в система за версиониране на данни (DVC, LakeFS).

9. Load in your training loop

   # PyTorch + WebDataset example
   import webdataset as wds, torch, torchvision, torchaudio
   
   def decode(sample):
       img = torchvision.io.decode_image(sample["jpg"], mode=torchvision.io.ImageReadMode.RGB)
       txt = torch.tensor([int(t) for t in sample["txt"].decode().split()], dtype=torch.long)
       wav, _ = torchaudio.load(io.BytesIO(sample["wav"]))
       return {"image": img, "caption": txt, "audio": wav}
   
   ds = (wds.WebDataset("s3://my-bucket/train-{00000..00999}.tar.zst")
         .decode("torchrgb")
         .map(decode)
         .batched(64)
         .prefetch(2))
   
   loader = torch.utils.data.DataLoader(ds, num_workers=8)
   for batch in loader:
       # feed to model …
       pass
   

Нарастващи тенденции и бъдеща устойчивост

Списък за готовност за продукция

• Файловата схема (.proto или Arrow schema) съхранена до данните.
• Всички шарди компресирани с бърз кодек (препоръчва се ZSTD‑L3).
• Размер на шарда между 256 MiB и 1 GiB.
• Манифестът включва контролна сума, брой записи, статистика за всеки шард и git hash на кода за предварителна обработка.
• Неизменимо контролиране на версии (DVC, LakeFS или подобно).
• Метрики за качество на данните записани за всеки шард.
• Проведен одит за поверителност (премахване на лични данни, по избор криптиране).
• Край‑до‑край тестов зареждач, който може да прочете случаен шард без грешки.
• README, който обяснява схемата, стъпките за предварителна обработка и как да се регенерират шарди.

Следвайки този план, вашите обучителни тръбопроводи ще останат бързи, евтини и възпроизведими — трите стълба, от които се нуждае всеки модерен LLM екип.

Tags: data‑engineering multi‑modal‑llm training‑pipelines
Slug: how-to-prepare-data-file-formats-for-ai-training