DeepSpeech/doc/Geometry.rst

64 строки
2.4 KiB
ReStructuredText
Исходник Постоянная ссылка Обычный вид История

2017-02-03 03:57:58 +03:00
Geometric Constants
===================
2017-01-03 17:43:13 +03:00
This is about several constants related to the geometry of the network.
2017-02-03 03:57:58 +03:00
n_input
-------
Each of the at maximum ``n_steps`` vectors is a vector of MFCC features of a
2017-01-03 17:43:13 +03:00
time-slice of the speech sample. We will make the number of MFCC features
dependent upon the sample rate of the data set. Generically, if the sample rate
is 8kHz we use 13 features. If the sample rate is 16kHz we use 26 features...
We capture the dimension of these vectors, equivalently the number of MFCC
2019-12-02 13:04:27 +03:00
features, in the variable ``n_input``. By default ``n_input`` is 26.
2017-01-03 17:43:13 +03:00
2017-02-03 03:57:58 +03:00
n_context
---------
2019-12-02 13:04:27 +03:00
As previously mentioned, the RNN is not simply fed the MFCC features of a given
time-slice. It is fed, in addition, a context of :math:`C` frames on
2017-01-03 17:43:13 +03:00
either side of the frame in question. The number of frames in this context is
2019-12-02 13:04:27 +03:00
captured in the variable ``n_context``. By default ``n_context`` is 9.
2017-01-03 17:43:13 +03:00
Next we will introduce constants that specify the geometry of some of the
non-recurrent layers of the network. We do this by simply specifying the number
2017-02-03 03:57:58 +03:00
of units in each of the layers.
2017-01-03 17:43:13 +03:00
2017-02-03 03:57:58 +03:00
n_hidden_1, n_hidden_2, n_hidden_5
----------------------------------
``n_hidden_1`` is the number of units in the first layer, ``n_hidden_2`` the number
of units in the second, and ``n_hidden_5`` the number in the fifth. We haven't
2017-01-03 17:43:13 +03:00
forgotten about the third or sixth layer. We will define their unit count below.
2019-12-02 13:04:27 +03:00
The RNN consists of an LSTM RNN that works "forward in time":
2017-01-03 17:43:13 +03:00
2017-02-03 03:57:58 +03:00
.. image:: ../images/LSTM3-chain.png
:alt: Image shows a diagram of a recurrent neural network with LSTM cells, with arrows depicting the flow of data from earlier time steps to later timesteps within the RNN.
2017-01-03 17:43:13 +03:00
The dimension of the cell state, the upper line connecting subsequent LSTM units,
2019-12-02 13:04:27 +03:00
is independent of the input dimension.
2017-01-03 17:43:13 +03:00
2017-02-03 03:57:58 +03:00
n_cell_dim
----------
2017-01-03 17:43:13 +03:00
Hence, we are free to choose the dimension of this cell state independent of the
2017-02-03 03:57:58 +03:00
input dimension. We capture the cell state dimension in the variable ``n_cell_dim``.
2017-01-03 17:43:13 +03:00
2017-02-03 03:57:58 +03:00
n_hidden_3
----------
2017-01-03 17:43:13 +03:00
The number of units in the third layer, which feeds in to the LSTM, is
2017-02-03 03:57:58 +03:00
determined by ``n_cell_dim`` as follows
.. code:: python
2019-12-02 13:04:27 +03:00
n_hidden_3 = n_cell_dim
2017-02-03 03:57:58 +03:00
2019-12-02 13:04:27 +03:00
n_hidden_6
2017-02-03 03:57:58 +03:00
-----------
2019-12-02 13:04:27 +03:00
The variable ``n_hidden_6`` will hold the number of characters in the target
2017-02-03 03:57:58 +03:00
language plus one, for the :math:`blank`.
2017-01-03 17:43:13 +03:00
For English it is the cardinality of the set
2017-02-03 03:57:58 +03:00
.. math::
\{a,b,c, . . . , z, space, apostrophe, blank\}
2017-01-03 17:43:13 +03:00
we referred to earlier.