CTCLoss

class torch.nn.CTCLoss(blank=0, reduction='mean', zero_infinity=False) [source]

Потеря Connectionist Temporal Classification.

Вычисляет потерю между непрерывной (неразбитой) временной серией и целевой последовательностью. CTCLoss суммирует вероятность возможных выравниваний входных данных к целевым данным, генерируя значение потери, которое является дифференцируемым по отношению к каждому узлу входных данных. Предполагается выравнивание входных данных к целевым «многие-к-одному», что ограничивает длину целевой последовательности так, чтобы она должна быть $\leq$ длине входных данных.

Параметры:

• blank (int, необязательно) – метка пропуска. Значение по умолчанию $0$.
• reduction (str, необязательно) – Указывает уменьшение, применяемое к выводу: 'none' | 'mean' | 'sum'. 'none': уменьшение не будет применено, 'mean': значения потерь на выходе будут разделены на длины целевых данных, а затем вычисляется среднее по партии. Значение по умолчанию: 'mean'
• zero_infinity (bool, необязательно) – Нулевое значение бесконечных потерь и соответствующих градиентов. Значение по умолчанию: False Бесконечные потери в основном возникают, когда входные данные слишком короткие, чтобы их можно было сопоставить с целевыми данными.

Форма:

• Log_probs: Тензор размера $(T, N, C)$ или $(T, C)$, где $T = \text{длина входных данных}$, $N = \text{размер пакета}$, и $C = \text{количество классов (включая пропуск)}$. Логарифмированные вероятности выходов (например, полученные с помощью torch.nn.functional.log_softmax()).
• Targets: Тензор размера $(N, S)$ или $(\operatorname{sum}(\text{target\_lengths}))$, где $N = \text{размер пакета}$ и $S = \text{максимальная длина целевых данных, если форма } (N, S)$. Представляет целевые последовательности. Каждый элемент в целевой последовательности — индекс класса. И индекс цели не может быть пустым (по умолчанию 0). В форме $(N, S)$ целевые данные заполняются до длины самой длинной последовательности и стекируются. В форме $(\operatorname{sum}(\text{target\_lengths}))$ целевые данные предполагаются без заполнения и конкатенируются в 1 измерении.
• Input_lengths: Кортеж или тензор размера $(N)$ или $()$, где $N = \text{размер пакета}$. Представляет длины входов (должны быть $\leq T$). И длины указываются для каждой последовательности для достижения маскирования при предположении, что последовательности заполняются до одинаковой длины.
• Target_lengths: Кортеж или тензор размера $(N)$ или $()$, где $N = \text{размер пакета}$. Представляет длины целей. Длины указываются для каждой последовательности для достижения маскирования при предположении, что последовательности заполняются до одинаковой длины. Если форма цели $(N,S)$, target_lengths фактически являются индексом остановки $s_n$ для каждой целевой последовательности, такой что target_n = targets[n,0:s_n] для каждой цели в пакете. Длины должны быть $\leq S$. Если цели заданы как 1-мерный тензор, который является конкатенацией отдельных целей, target_lengths должны складываться в общую длину тензора.
• Выход: скаляр. Если reduction равен 'none', то $(N)$, если вход закодирован в пакет или $()$, если вход не закодирован в пакет, где $N = \text{размер пакета}$.

Примеры:

>>> # Target are to be padded
>>> T = 50      # Input sequence length
>>> C = 20      # Number of classes (including blank)
>>> N = 16      # Batch size
>>> S = 30      # Target sequence length of longest target in batch (padding length)
>>> S_min = 10  # Minimum target length, for demonstration purposes
>>>
>>> # Initialize random batch of input vectors, for *size = (T,N,C)
>>> input = torch.randn(T, N, C).log_softmax(2).detach().requires_grad_()
>>>
>>> # Initialize random batch of targets (0 = blank, 1:C = classes)
>>> target = torch.randint(low=1, high=C, size=(N, S), dtype=torch.long)
>>>
>>> input_lengths = torch.full(size=(N,), fill_value=T, dtype=torch.long)
>>> target_lengths = torch.randint(low=S_min, high=S, size=(N,), dtype=torch.long)
>>> ctc_loss = nn.CTCLoss()
>>> loss = ctc_loss(input, target, input_lengths, target_lengths)
>>> loss.backward()
>>>
>>>
>>> # Target are to be un-padded
>>> T = 50      # Input sequence length
>>> C = 20      # Number of classes (including blank)
>>> N = 16      # Batch size
>>>
>>> # Initialize random batch of input vectors, for *size = (T,N,C)
>>> input = torch.randn(T, N, C).log_softmax(2).detach().requires_grad_()
>>> input_lengths = torch.full(size=(N,), fill_value=T, dtype=torch.long)
>>>
>>> # Initialize random batch of targets (0 = blank, 1:C = classes)
>>> target_lengths = torch.randint(low=1, high=T, size=(N,), dtype=torch.long)
>>> target = torch.randint(low=1, high=C, size=(sum(target_lengths),), dtype=torch.long)
>>> ctc_loss = nn.CTCLoss()
>>> loss = ctc_loss(input, target, input_lengths, target_lengths)
>>> loss.backward()
>>>
>>>
>>> # Target are to be un-padded and unbatched (effectively N=1)
>>> T = 50      # Input sequence length
>>> C = 20      # Number of classes (including blank)
>>>
>>> # Initialize random batch of input vectors, for *size = (T,C)
>>> input = torch.randn(T, C).log_softmax(2).detach().requires_grad_()
>>> input_lengths = torch.tensor(T, dtype=torch.long)
>>>
>>> # Initialize random batch of targets (0 = blank, 1:C = classes)
>>> target_lengths = torch.randint(low=1, high=T, size=(), dtype=torch.long)
>>> target = torch.randint(low=1, high=C, size=(target_lengths,), dtype=torch.long)
>>> ctc_loss = nn.CTCLoss()
>>> loss = ctc_loss(input, target, input_lengths, target_lengths)
>>> loss.backward()

Ссылка:

A. Graves et al.: Connectionist Temporal Classification: Labelling Unsegmented Sequence Data with Recurrent Neural Networks: https://www.cs.toronto.edu/~graves/icml_2006.pdf

Примечание

Для использования CuDNN необходимо выполнить следующие условия: targets должно быть в конкатенированном формате, все input_lengths должны быть T. $blank=0$, target_lengths $\leq 256$, целочисленные аргументы должны иметь тип torch.int32.

Регулярная реализация использует (более распространённый в PyTorch) тип torch.long.

Примечание

В некоторых случаях при использовании CUDA бэкенда с CuDNN этот оператор может выбрать недетерминированный алгоритм для повышения производительности. Если это нежелательно, вы можете попытаться сделать операцию детерминированной (возможно, с потерей производительности) путём установки torch.backends.cudnn.deterministic = True. Обратитесь к примечаниям по Воспроизводимости для получения дополнительной информации.