Skip to content

RiTA-nlp/ITALIC

Repository files navigation

ITALIC: An ITALian Intent Classification Dataset

ITALIC is an intent classification dataset for the Italian language, which is the first of its kind. It includes spoken and written utterances and is annotated with 60 intents. The dataset is available on Zenodo and connectors ara available for the HuggingFace Hub.

official paper HuggingFace Dataset

Latest Updates

Table of Contents

Data collection

The data collection follows the MASSIVE NLU dataset which contains an annotated textual dataset for 60 intents. The data collection process is described in the paper Massive Natural Language Understanding.

Following the MASSIVE NLU dataset, a pool of 70+ volunteers has been recruited to annotate the dataset. The volunteers were asked to record their voice while reading the utterances (the original text is available on MASSIVE dataset). Together with the audio, the volunteers were asked to provide a self-annotated description of the recording conditions (e.g., background noise, recording device). The audio recordings have also been validated and, in case of errors, re-recorded by the volunteers.

All the audio recordings included in the dataset have received a validation from at least two volunteers. All the audio recordings have been validated by native italian speakers (self-annotated).

Dataset

The dataset is available on the Zenodo. It is composed of 3 different splits:

  • massive: all the utterances are randomly shuffled and divided into 3 splits (train, validation, test).
  • hard_speaker: the utterances are divided into 3 splits (train, validation, test) based on the speaker. Each split only contains utterances from a pool of speakers that do not overlap with the other splits.
  • hard_noisy: the utterances are divided into 3 splits (train, validation, test) based on the recording conditions. The test split only contains utterances with the highest level of noise.

Each split contains the following annotations:

  • utt: the original text of the utterance.
  • audio: the audio recording of the utterance.
  • intent: the intent of the utterance.
  • speaker: the speaker of the utterance. The speaker is identified by a unique identifier and has been anonymized.
  • age: the age of the speaker.
  • is_native: whether the speaker is a native italian speaker or not.
  • gender: the gender of the speaker (self-annotated).
  • region: the region of the speaker (self-annotated).
  • nationality: the nationality of the speaker (self-annotated).
  • lisp: any kind of lisp of the speaker (self-annotated). It can be empty in case of no lisp.
  • education: the education level of the speaker (self-annotated).
  • environment: the environment of the recording (self-annotated).
  • device: the device used for the recording (self-annotated).

Usage

The dataset can be loaded using the datasets library. You need to install the following dependencies:

pip install datasets
pip install librosa
pip install soundfile

Then, you can load the dataset as follows:

from datasets import load_dataset

# Please be sure to use use_auth_token=True and to set the access token
# using huggingface-cli login
# or follow https://huggingface.co/docs/hub/security-tokens 

# configs "hard_speaker" and "hard_noisy" are also available (to substitute "massive")
italic = load_dataset("RiTA-nlp/ITALIC", "massive", use_auth_token=True) 
italic_train = italic["train"]
italic_valid = italic["validation"]
italic_test  = italic["test"]

The dataset has been designed for intent classification tasks. The intent column can be used as the label. However, the dataset can be used for other tasks as well.

  • Intent classification: the intent column can be used as the label.
  • Speaker identification: the speaker column can be used as the label.
  • Automatic speech recognition: the utt column can be used as the label.
  • Accent identification: the region column can be used as the label.

For more information about the dataset, please refer to the paper.

Models used in the paper

Hardware settings

All experiments were conducted on a private workstation with Intel Core i9-10980XE CPU, 1 $\times$ NVIDIA RTX A6000 GPU, 64 GB of RAM running Ubuntu 22.04 LTS.

Parameter settings

The parameters used for the training of the models are set to allow a fair comparison between the different models and to follow the recommendations of the related literature. The parameters are summarized in the following table:

Model Task Parameters Learning rate Batch size Max epochs Warmup Weight decay Avg. training time Avg. inference time
facebook/wav2vec2-xls-r-300m SLU 300M 1e-4 128 30 0.1 ratio 0.01 9m 35s per epoch 13ms per sample
facebook/wav2vec2-xls-r-1b SLU 1B 1e-4 32 30 0.1 ratio 0.01 21m 30s per epoch 29ms per sample
jonatasgrosman/wav2vec2-large-xlsr-53-italian SLU 300M 1e-4 128 30 0.1 ratio 0.01 9m 35s per epoch 13ms per sample
jonatasgrosman/wav2vec2-xls-r-1b-italian SLU 1B 1e-4 32 30 0.1 ratio 0.01 21m 30s per epoch 29ms per sample
ALM/whisper-it-small-augmented ASR 224M 1e-5 8 5 500 steps 0.01 26m 30s per epoch 25ms per sample
EdoAbati/whisper-medium-it-2 ASR 769M 1e-5 8 5 500 steps 0.01 49m per epoch 94ms per sample
EdoAbati/whisper-large-v2-it ASR 1.5B 1e-5 8 5 500 steps 0.01 1h 17m per epoch 238ms per sample
bert-base-multilingual-uncased NLU 167M 5e-5 8 5 500 steps 0.01 1m 22s per epoch 1.5ms per sample
facebook/mbart-large-cc25 NLU 611M 5e-5 8 5 500 steps 0.01 7m 53s per epoch 4.7ms per sample
dbmdz/bert-base-italian-xxl-uncased NLU 110M 5e-5 8 5 500 steps 0.01 1m 30s per epoch 1.4ms per sample
morenolq/bart-it NLU 141M 5e-5 8 5 500 steps 0.01 1m 54s per epoch 1.9 ms per sample

In all cases, we opted for the AdamW optimizer. All experiments were run on a single NVIDIA A6000 GPU.

SLU intent classification

The models used in the paper are available on the Hugging Face Hub.

ASR

The models used in the paper are available on the Hugging Face Hub.

NLU intent classification

The models used in the paper are available on the Hugging Face Hub.

Citation

If you use this dataset in your research, please cite the following paper:

@inproceedings{koudounas23_interspeech,
  author={Alkis Koudounas and Moreno {La Quatra} and Lorenzo Vaiani and Luca Colomba and Giuseppe Attanasio and Eliana Pastor and Luca Cagliero and Elena Baralis},
  title={{ITALIC: An Italian Intent Classification Dataset}},
  year=2023,
  booktitle={Proc. INTERSPEECH 2023},
  pages={2153--2157},
  doi={10.21437/Interspeech.2023-1980}
}

License

The dataset is licensed under the Creative Commons Attribution 4.0 International License.

About

ITALIC: An ITALian Intent Classification Dataset

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •