用Label Studio标注训练数据

这个方案在项目中已经稳定运行了一段时间，整理出来分享。有些细节可能因版本不同有差异，注意看注释。

先搞清楚原理

Label Studio的基本概念其实不复杂。关键在于理解特征工程这个环节。看下面这段代码，逻辑就很清楚了：

import torch
import torch.nn as nn

class Model_256(nn.Module):
    def __init__(self, input_dim=256, hidden_dim=512, num_classes=10):
        super().__init__()
        self.encoder = nn.Sequential(
            nn.Linear(input_dim, hidden_dim),
            nn.SiLU(),
            nn.Dropout(0.1),
            nn.Linear(hidden_dim, hidden_dim // 2),
            nn.SiLU(),
        )
        self.classifier = nn.Linear(hidden_dim // 2, num_classes)

    def forward(self, x):
        features = self.encoder(x)
        return self.classifier(features)

model = Model_256()
total_params = sum(p.numel() for p in model.parameters())
print(f"参数量: {total_params:,}")
x = torch.randn(32, 256)
out = model(x)
print(f"输入: {x.shape} -> 输出: {out.shape}")

实际怎么用

实际编码中，Label Studio的用法可以简化。下面是我在项目中用的写法，经过几轮迭代优化过：

import faiss
import numpy as np

dim = 768
n_vectors = 100000
vectors = np.random.randn(n_vectors, dim).astype("float32")
faiss.normalize_L2(vectors)

index = faiss.IndexFlatIP(dim)
index.add(vectors)

query = np.random.randn(1, dim).astype("float32")
faiss.normalize_L2(query)
k = 10
distances, indices = index.search(query, k)
print(f"Top-{k} 相似向量索引: {indices[0]}")
print(f"对应相似度: {distances[0]}")

进阶一点的用法

进阶一点的话，Label Studio还有梯度反向传播这个方向可以探索。原理不展开，直接看关键代码：

容易踩的坑

用Label Studio的时候有几个常见的错误写法。比如特征工程没有正确处理，或者梯度反向传播的顺序搞反了。下面列出几个典型问题和修复方式。

写在最后

整体来看，Label Studio用起来不算复杂，关键是理解梯度反向传播这个核心机制。建议先跑通基础示例，再逐步加功能。有问题可以看官方文档，社区也比较活跃。