Git Product home page Git Product logo

naver_ai_hackathon_speech's Introduction

NAVER AI HACKATHON 2019 - Speech To Text

We participated in NAVER_AI_HACKTHON 2019 and ranked 9th(77.0155) as a two-man team(Hangbok Coding, 행복코딩).

Final Leaderboard

finale-leader-board

Features

  • Convolution feature extraction (deepspeech style)
  • Seq2Seq (bidirectional GRU encoder, unidirectional GRU decoder with Bahdanau Attention)
  • Augmentation with log mel spectrogram (SpecAugment)
  • Beam-search
  • Data preprocessing (deleted blank and special characters)
  • Label Smoothing
  • Multi step learning rate
  • Ensemble (but not used for best model)

Data

Naver has released approximately 100 hours of 16kHz Korean speech data.

Data format

  • Audio data : 16bit, mono 16k sampling PCM, WAV audio
  • Target script : Korean characters are converted to the indices according to the 'hackathon.labels' file
    "네 괜찮습니다." => "715 662 127 76 396 337 669 662"

Dataset folder structure

* DATASET-ROOT-FOLDER
|--train
   |--train_data
      +--data_list.csv
      +--a.wav, b.wav, c.wav ...
   +--train_label
  • data_list.csv

    <wav-filename>,<script-filename>
    wav_001.wav,wav_001.label
    ...
    
  • train_label

    <filename>,<script labels>
    wav_001,628 9 625 662 408 690 2 125 71 662 220 630 610 749 62 661 123 662
    ...
    

How to RUN

Docker

$ docker build -t model:0.0 .
$ docker run -i --name model model:0.0
$ docker exec -i -t model /bin/bash
$ ./run.sh

NSML

Login with nsml first, and run commands as follows:

$ ./run.sh # for local training
$ ./run_nsml.sh  # for NSML training
$ nsml submit (sessionName) (checkpoint) # for submit

Hyperparameters

Help default
hidden size of model 256 --hidden_size
size of embedding dimension 64 --embedding_size
number of layers of encoder 4 --encoder_layer_size
number of layers of decoder 3 --decoder_layer_size
batch size 32 --batch_size
initial learning rate 1e-04 --lr
teacher forcing 0.5 --teacher_forcing
maximum characters of sentence 80 --max_len

Reference

  1. Listen, Attend and Spell
  2. Label Smoothing
  3. SpecAugment
  4. Naver's baseline code

License

Copyright 2019 Hangbok Coding.

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and
associated documentation files (the "Software"), to deal in the Software without restriction, including
without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to
the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial
portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE
OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

naver_ai_hackathon_speech's People

Contributors

elzino avatar psi9730 avatar clovaaiadmin avatar bluebrush avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.