Next Article in Journal
Air Quality and Comfort Characterisation within an Electric Vehicle Cabin in Heating and Cooling Operations
Previous Article in Journal
A Lightweight Passive Human Tracking Method Using Wi-Fi
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Gaze Tracking Based on Concatenating Spatial-Temporal Features

1
Department of Computer and Communication Engineering, Ming Chuan University, Taoyuan 333, Taiwan
2
College of Artificial Intelligence, Yango University, Fuzhou 350015, China
*
Author to whom correspondence should be addressed.
Sensors 2022, 22(2), 545; https://doi.org/10.3390/s22020545
Submission received: 1 December 2021 / Revised: 3 January 2022 / Accepted: 8 January 2022 / Published: 11 January 2022
(This article belongs to the Section Biomedical Sensors)

Abstract

:
Based on experimental observations, there is a correlation between time and consecutive gaze positions in visual behaviors. Previous studies on gaze point estimation usually use images as the input for model trainings without taking into account the sequence relationship between image data. In addition to the spatial features, the temporal features are considered to improve the accuracy in this paper by using videos instead of images as the input data. To be able to capture spatial and temporal features at the same time, the convolutional neural network (CNN) and long short-term memory (LSTM) network are introduced to build a training model. In this way, CNN is used to extract the spatial features, and LSTM correlates temporal features. This paper presents a CNN Concatenating LSTM network (CCLN) that concatenates spatial and temporal features to improve the performance of gaze estimation in the case of time-series videos as the input training data. In addition, the proposed model can be optimized by exploring the numbers of LSTM layers, the influence of batch normalization (BN) and global average pooling layer (GAP) on CCLN. It is generally believed that larger amounts of training data will lead to better models. To provide data for training and prediction, we propose a method for constructing datasets of video for gaze point estimation. The issues are studied, including the effectiveness of different commonly used general models and the impact of transfer learning. Through exhaustive evaluation, it has been proved that the proposed method achieves a better prediction accuracy than the existing CNN-based methods. Finally, 93.1% of the best model and 92.6% of the general model MobileNet are obtained.

1. Introduction

Most people rely on vision in receiving information. Fixation is the main way to receive visual information. Observing eye movements can observe the cognitive process and concentration [1]. Therefore, tracking the estimated gaze is considered as the most effective way to evaluate visual behavior. The technology of gaze tracking has been gradually applied in different fields in recent years, such as Human–Computer Interaction (HCI), audience gaze behavior analysis, advertising analysis, gaze information collection and analysis on webpages, medical disease interpretation training [2,3].
In recent years, due to the development of machine learning technology, many studies have used deep learning training methods to improve the accuracy of gaze point estimation [4,5]. Since the Convolutional Neural Network (CNN) excels in many computer vision tasks, it is also used for gaze point estimation [2,6,7]. For CNN-based gaze point estimation, the image is selected as the training and predictive data with the extracted spatial features. Deep learning requires a lot of data for training to avoid overfitting. There have been several publicly available datasets for gaze estimation [8,9,10,11,12,13]. Many training samples for gaze estimations are mainly image-based [8,9,10]. Few samples are video-based collected from people viewing dynamic stimuli such as watching YouTube and Netflix [11,12,13].
In terms of appearance-based gaze estimation, the authors of [8] constructed a LeNet-inspired CNN to learn the mapping between 2D head angle, eye image and output gaze angle. Another study [9] constructed a dataset containing 64,000 appearance-based images from 50 subjects. In a controlled environment where the subject’s head was fixed, images of the subject were collected from a fixed distance. These images possessed dense sampling of gaze angles. Krafka et al. [10] used GazeCapture to obtain a large amount of gaze data using mobile phones as vehicles, and used these data as training data for their network, iTracker, a convolutional neural network for eye tracking. In terms of providing video datasets, the datasets of [11] contained images recorded by color and depth cameras with head pose of the participant, visual target, and recording conditions. However, the dataset only recorded the data of three subjects under two different environmental conditions. The Discrete screen target drew a small circle evenly every 1.1 s at a random position on the screen. The Continuous screen target moved the circle along a random trajectory for 2 s. In [9,11], although all the above datasets captured a large number of head poses and appearances, the experiments were conducted in a strictly controlled manner without changes in body postures. A strictly controlled experimental environment was not a practical live condition in which a subject’s head and body moved naturally influenced by the emotional changes during exploring media contents. The research reported in [12] collected the gaze point, head position and direction in the three-dimensional space as the data when asking participants to look at the predetermined target on the monitor. The test targets were arranged as 3 × 3 and 4 × 4 fixed points on the screen. The participants were instructed to gaze from the fixed point to the center or from the center point to the target point. The study of [14] estimated the gaze on a tablet. The goal was to perform an uncalibrated gaze estimation with the front camera of the tablet, where the posture and method of holding the tablet were not limited. In [12,13], both articles traced the gazed area of interest and gaze patterns with small circles as the test target object, regardless of static graphics or dynamic movies.
In our previous research, three types of fed images and varied network architectures are proposed to estimate the gaze point and compare the efficiency on the architecture of applying the different convolutional layers, using batch normalization (BN) and applying the global average pooling layer (GAP) instead of the fully connected layer [15]. The input model of the face image is the best output result. Furthermore, there is a correlation between time and consecutive gaze positions in visual behaviors when people watch a dynamic material (such as a movie). Usually, the consecutive gaze positions are not too far apart. Therefore, this paper proposes to use the exploration results as the basis for the gaze point estimation design. In addition to the spatial feature, the temporal features are considered to improve the accuracy. The traditional neural network models for gaze estimation use a single image as the input training data but rarely use videos. If the video can be used for predicting, the effect can be better achieved on the gaze estimation through the context of the time series in the video.
Unlike traditional neural networks of which input data are independent from output data, a time-related machine learning network such as the Recurrent Neural Network (RNN) uses the output of the previous step as the input of the current step. However, the performance of RNN in long-term memory is not as expected, and Long Short-Term Memory (LSTM) is designed to improve the defect of the long-term memory [16,17]. LSTM was proposed by Hochreiter and Schmidhuber [16] in 1997 to solve the gradient descent problem of standard RNN. Since the standard RNN has only a simple structure, it is difficult to capture long-term memories. Therefore, LSTM introduces three gates that control memory in the neural unit to control the retention of memory: Forget Gate, Input Gate, and Output Gate.
LSTM has been applied in many fields, such as language modeling, text classification, activity recognition, acoustic modeling and stock market prediction, etc. LSTM is more commonly used early in stock returns prediction [18]. There are also studies using LSTM to explore time-sequential features in videos. For example, there are some studies using CNN and LSTM to propose some methods for action recognition [19,20,21,22,23,24]. The gaze-based early prediction of the user’s intention is proposed by employing a combination of CNN and LSTM [25]. By exploring literature studies, it rarely can be found combing both extracted spatial and temporal features at the same time for gaze estimation. For the research topic on gaze estimation, the available features for the training samples are mostly the extractions of eyes and face images [15]. Based on the above exploration, this research article hence proposes to concatenate spatial and temporal features and has named it as CCLN (CNN Concatenating LSTM Network) to improve the performance of gaze estimation in the case of time-series videos as the input training data.
Many research works focus on network architecture to improve performance or reduce complexity, such as GAP and BN [26,27]. In order to reduce the parameters of convolutional neural networks and avoid overfitting, the GAP layer is proposed to replace the fully connected layer [26]. The difference between the GAP layer and the fully connected layer is that the average value of each feature map of the last layer is taken out. Ioffe et al. [27] proposed the method of batch normalization to overcome the problem of hard-to-train models. The main idea is by making normalization a part of the model architecture and performing the normalization for each training mini-batch. The benefits of BN include improving the learning speed, reducing the dependence on initial parameters, and eliminating the need for dropout layer in some cases. Therefore, in this paper, the proposed model can be optimized by exploring the influence of GAP and BN on CCLN.
There are a few research works on gaze estimation that use videos based on temporal features as the input training data. This research article evaluates the performance of different network architectures and hyperparameter adjustments considering both spatial and the temporal features. Conventionally, prediction accuracy is often used as a metric to evaluate classifiers. However, the complexity of the model is also a very important design factor, because it will affect the calculation load and implementation cost of the system. Thus, in this research, several prediction metrics are used, including network complexity, accuracy, loss, recall, F1 score and precision for the evaluation and comparison of the networks. The major contributions of this work are summarized as follows:
  • In addition to model architecture, it is generally believed that larger amounts of training data will lead to better models. We propose a method for constructing a video dataset for gaze point estimation based on concatenating spatial and temporal futures.
  • We propose a simple, yet effective CNN Concatenate LSTM network (CCLN), to explore the performance of gaze estimation in the case of time-series videos as the input training data. Through exhaustive evaluation, we prove that the proposed method is achieving better prediction accuracy than the existing CNN-based methods.
  • This research takes account of the studied effect of the batch normalization and global average pooling methods into the designed network on better prediction accuracy and reducing complexity.
  • The effectiveness of different commonly used general models and the impact of transfer learning are studied.
The remainder of this paper is organized as follows. Section 2 first gives an overview of the proposed method, describes the data pre-processing algorithm, and then, the design of CCLN. The numerical analysis and performance comparison are given in Section 3. Finally, the conclusion is drawn in Section 4.

2. Proposed Concatenating Spatial and Temporal Features Method

In this section, first, an overview of the proposed concatenating spatial and temporal features method is stated, showing and explaining the system architecture. Second, the descriptions on the data pre-processing, including synchronizing data collection, labeling and building dataset are addressed. Last, the design of CCLN consisting of three functional blocks, CNN Block, LSTM Block and Output Block, is presented.

2.1. System Overview

Figure 1 shows the system architecture for gaze point estimation based on concatenating spatial and temporal futures, including Offline Training and Online Prediction procedures. There are three function blocks consisting of the proposed Dataset, CCLN Gaze Estimation Network, and Loss Calculation. In the Online Prediction procedure, the prediction model is transferred when the Offline Training procedure is terminated. In our previous research [15], the input model of the face image was the best output result; therefore, this study continues to use such type of images as the training and prediction data.
In the past, video was rarely used as the input data for gaze point estimation. This paper proposes to record facial images while participants watching a movie, supplemented by the gaze point coordinates labelled by the eye tracker, to create a dataset with time-series-synchronized data. This dataset will be used as the training sample of the CCLN proposed in this paper. CCLN Gaze Estimation Network is designed for concatenating spatial and temporal features. In the prediction procedure, video (i.e., Stimulus video) is used as an estimation unit.
This paper mainly completes several tasks including the following:
  • Synchronized data collection and automatic labeling method
Both the quantity and the content of training data are important factors that affect the final performance of machine learning. Data labeling is a heavy task. Designing an automatic labeling method can speed up and increase the construction of a training dataset.
2.
Constructing a dataset by watching movies
There are some public datasets built for viewing moving objects, mainly known objects such as small circles. They made participants gaze on the small circle on the screen, and then mark the connection coordinating this known object and the eye or face image. However, the visual behavior occurred in this kind of experimental environment setting is very different from that of generally watching a movie. There are few datasets that use videos (face) as the training data, so it is necessary to build a suitable time-series-related dataset. In this paper, the proposed method is to obtain the images taken by a webcam when participants viewing each movie frame. Further labeling the image to coordinate with the gaze point estimated by the infrared eye tracker to build a dataset of the synchronized time-series. However, such a taken image is still too messy and not suitable for direct use as a training sample, so the image needs to be cropped to a fixed size. In addition, a method is proposed to compose the taken images into a video with the consideration of an appropriate length.
3.
Designing the network architecture of CCLN
Other than using the eye image as the main extraction feature for gaze estimation, using the face image can overcome the influence caused by the head movement. Therefore, the training of CCLN will take the built video dataset as the training sample with the face image as the input image. Moreover, a complicated network architecture will easily lead to overfitting when the input image consists of few features. Hence, the number of LSTM layers, batch normalization and global average pooling are used to explore a model to obtain a balance between better prediction accuracy and reducing complexity. The performance, including accuracy, loss, recall, F1 score and precision, etc., are computed.
4.
Studying the performance of commonly used models and the impact of transfer learning
This paper evaluates the performance of CNN when it is replaced by commonly used network models, including VGG16, VGG19, ResNet50, DenseNet121 and MobileNet. In addition, whether or not the effect of trained weights is initially used during training is evaluated, as a way to explore the generality of the network architecture proposed in this paper.

2.2. Synchronizing Data Collection, Labeling and Building Dataset

In order to obtain datasets of eye and face images for labeling, it is common to manage the viewable screen area into M × N regions with showing a predetermined target to be viewed, the target usually as a circle appearing in the center or moving toward the center of each region area in a random manner [12,13]. The eye images and the face images tracing the target to be viewed are recorded and marked during the viewing process. There are two problems with the abovementioned obtainment of the datasets. Firstly, the data obtained are limited to the center position of gazed region blocks, and the amount of data is relatively insufficient.
Secondly, by observing the results of static images [28,29], which have been marked with the gaze points, it can be found that the gaze points are actually concentrated on the objects of interest. For example, it can be proved from Figure 2 [28] that the position of the object in the image will affect the visual behavior, which will affect the area of interest for viewing, so the abovementioned obtainment of the datasets [12,13] is not favorable. Besides, although datasets such as [28,29] contain gaze point information, they lack the eye image at this time and cannot be directly used for training.
In addition, we have conducted an experiment on the gaze behavior of the participants while watching the arranged video regarding to the pangolin object. Figure 3 shows the examples that the visual behavior exhibits personal preferences with distinctly different distributions of gazes. It exhibits the result of comparing the gaze points of the different participants during frame 2387 to frame 2598 to express the attentions. Comparing to participant No. 2, the distributions of gaze points of participants No. 6 and No. 7 are more aggregated. Even though the areas of interest are different, all of the three participants have attended the pangolin object. This result also demonstrates that the gaze point is clustered and time-related.
For obtaining the training dataset, this research paper proposes to adopt an eye tracker to record gaze points of the participants while watching the video of general films. There are currently some eye trackers that use infrared light to detect gaze points with acceptable accuracy, and they have been adopted by some studies [28]. For the eye tracker to achieve higher detection accuracy, a calibration process needs to be conducted prior to engaging the participant in the recording process initiated by the eye tracker. One should be also aware that the functionable distance for an eye tracker to detect gazes is relatively short as well as sever head and body movements might have impacts on detection accuracy. This infrared type of eye tracker is an acceptable auxiliary tool for acquiring gaze points to implement the proposed automatic labeling. Furthermore, a webcam along with the eye tracker is used for acquiring face images at the meantime to build the corresponding eye and face datasets.
To achieve the automatically labeling data for training, the acquired data of gaze points detected by the eye tracker and face images captured by the webcam must first be synchronized, in terms of the timestamps and the corresponding frame numbers. Therefore, this research paper designs a method for obtaining the synchronized data. In order to keep simultaneously with the frame rate of the video, the webcam is set to capture the face image of the participant while watching the video with one shot per frame. That is, the frequency of taking face images is the same as the frequency of the video frame rate. In the meantime, the (x, y) coordinates and time (frame number) of the current gaze point taken by eye tracker are recorded. The synchronization is implemented by using the frame number as the unit of time to associate the data between the coordinates of a gaze point and a captured face image. In order to achieve the purpose of automatically labeling data, in the program development, we first play the video and record the number of the frame, then capture the face image and get the gaze point detected by the gaze tracker at this time. Finally, the number of the frame, gaze point and face image serial number are combined into a record of data for storage. When training the input video, the corresponding relevant annotation data can be obtained at the same time.
Furthermore, the face images obtained by the webcam need to be cropped prior to be used as the training samples. Generally, facial recognition using facial landmarks for calibrating is to adjust the face to the center of the picture, the eyes of the face are on the horizontal line, and the face size of each picture is scaled to the same size. This study uses facial landmarks for facial recognition and using the center of the frame as the alignment base to crop the obtained face images. Since the heads and bodies of participants might move in pitch, roll and yaw with the distance changes from the camera or participants posse different eye heights, there are different image patterns of face and eye when participants look at the same position. These variations of different image patterns corresponding to the same gaze point should be regarded as visual behavior characteristics. If the obtained face image is cropped with the face as the alignment center, part of the features will be lost. To retain these features, the obtained face image will be cropped relative to the center of the frame.
It is observed that when gaze points gather on a certain object of interest in a short time, there will be no large-scale movement of adjacent positions. When the object of interest is changed, gaze points may shift away to another interested object with more movements. Based on the above observation on gathering effects and distinguished movements of shifting gaze points associated to the object of interests, this study proposes a method to composite the captured corresponding face images into the video as the training dataset. Gaze is dynamic and the probability of having consecutive gaze points with the same coordinates is extremely low. The face images are encoded into video to build the dataset with the following rule. This study presents the concept of a B × B macroblock, with size of width B pixels and height B pixels, as a basis for composing the training data. If the nth frame is the first frame of the video to be produced, we calculate the distance of the gaze points between the nth frame and the (n + 1)th frame. If the distance is within B/2 pixels, we record this frame and continue to calculate the next frame until it exceeds B/2 pixels, and then proceed to the next calculation procedure. The face images mapping to frames in the same recording are converted into a video, i.e., a training or test data in the dataset. The face images of the gaze points in a macroblock of B × B pixels are converted into a training data. The lengths of the composed training videos will adversely affect the training results. The built dataset is randomly split into 0.7, 0.2, and 0.1 for training, testing, and evaluation sets, respectively. A built dataset is a labelled folder storing data collected from each participant, respectively. Prior to performing training and testing the model, data in a respective dataset folder are randomly selected and split into 7:2:1 for training, testing, and evaluation sets. The approach of splitting the datasets into training and testing datasets can be reproducible.
The experimental tools for collecting data and 6 selected videos to watch are the same as used in [15], including a 22-inch with 16:9 screen is used as the video playback device, Logitech C920r HD Pro Webcam and Tobii Eye Tracker 4C, respectively, for capturing face image and getting the gaze point. The participants watched 6 selected videos, each with a size of 1920 × 1080 pixels and a frame rate of 24 frames per second, played in a randomly selected manner. The detailed information of the videos is shown in Table 1. The experiment and data are newly conducted in accordance to this research purpose. In this study experiment, there are 28 participants and face images of them are collected. However, the data type used in this paper is different. The input data use video instead of images, whereas [15] used images.
In the case of setting B to 30, after the production of the video, the video length distribution is less than 5 s, and 90% is within 1 s. Therefore, the threshold is set as 3 s, and if it exceeds, it will be divided into 2 equal segments. The dataset contains 1017 videos labeled with block numbers out of the 37 blocks proposed by [15]. Figure 4 shows the distribution results after the videos are labeled as blocks. According to the proposed video splitting method, the length of each video is controlled within 3 s and the distribution is shown in Table 2. When the participant is watching the video, the gaze points are usually in the area near the center of the screen, resulting in an uneven data distribution. The data augmentation method of brightness adjustments will be performed to expand the number of videos in blocks which contain fewer videos than the average number of videos in all blocks. Thus, the blocks with a lower number of videos can get enough data, and finally, there are 1517 videos after data augmentation, as shown in Figure 5.

2.3. The Proposed CCLN Gaze Estimation Network

Figure 6 shows the architecture of CCLN Gaze Estimation Network consisted of 3 functional blocks, CNN Block, LSTM Block and Output Block. The architecture of CNN Block includes 12 layers of convolutional layers and 4 layers of Max Pooling. The number of convolutional layers is connected in series with 2, 2, 4, 4, and the middle is connected with the Max Pooling layer. After each convolutional layer in the model, batch normalization (BN) and Relu functions are added to reduce the computational complexity and are illustrated as light green and gray, respectively. The architecture of this CNN block is similar to that of [15]. The difference is that there are no global average pooling (GAP) and Softmax layers, so it can have a better effect on transfer learning. The dataset built in this paper is randomly split into training, testing, and evaluation sets. According to the system architecture shown in Figure 1, in the Offline Training state, a video in the training set is randomly selected and decoded into frames to input CCN in CCLN for training. For each built dataset, data collected from per participant, respectively, data in a respective dataset folder are randomly selected and split into 7:2:1 for training, testing, and evaluation sets. During performing Offline Training state, a video will be randomly selected from the training set, and then, the next one will be randomly selected, so forth and so on. Data for performing testing and evaluation are as well randomly selected with the same approach. In addition, the technique of early stopping is applied to end the training procedure. Then, the testing set is used to validate the performance of the trained model. Finally, the evaluation set is used to evaluate the effectiveness of the trained model including accuracy, loss, F1 score, recall and precision in the Online Prediction state. The LSTM Block analyzes the system performance including the following different conditions: the number of LSTM layers, the comparison of unidirectional and bidirectional LSTM architectures, and whether to use Dropout, BN. Usually, the Output Block includes fully connected (FC) and Softmax layers, the issue of the impact of replacing fully connected layer with GAP will be evaluated and presented in Section 3.1.2.

3. Numerical Analysis

In this work, we evaluate the performance for CCLN Gaze Estimation Network including layers of LSTM, unidirectional and bidirectional LSTM, to use Dropout and BN, GAP. In addition, the comparison of various models is presented.

3.1. The Evaluation for LSTM Block

3.1.1. Layers of LSTM, Directional LSTM

The multi-layer LSTM network is used to extract temporal features layer by layer and improves the robustness of the model [30]. That the multi-layer LSTM can be stacked and temporally concatenated to form more complex structures has been applied to solve the real-life sequence modeling problem [31]. Several researchers made use of the bidirectional LSTM to improve the performance [32,33]. The bidirectional LSTM is the most performant to LSTM and GRU classifiers for fall event detection [32]. In [33], they apply bidirectional LSTM for real-time anomaly detection in surveillance networks. Therefore, in this paper, we study the effects of various architecture issues on the design of networks including bidirectional LSTM and number of layers in estimation the gaze network. Table 3 shows the types of LSTM Block architectures to be evaluated.
  • Type 1 unidirectional
Compared with [15], the accuracy of proposed CCLN is higher than [15] in the face scheme as shown in Table 4, with an approximately 4% performance improvement. The results show that using video as the input training data and using network of CNN concatenating LSTM can improve the accuracy for gaze estimation by 3.36%.
B.
Type 2 bidirectional
Compared with the unidirectional LSTM architecture, the bidirectional LSTM enables the model to make more accurate classifications as shown in Table 5, with an approximately 2.69% performance improvement.
C.
Type 3 unidirectional multi-layers
The next experiment is to compare the results of multiple layers for the unidirectional LSTM architecture as shown in Figure 7. When the number of layers is increased from 1 to 2, the accuracy is increased to 0.911. However, when it increases to 3 or 4, the accuracy reduces to 0.874 and 0.87, respectively. From the results, it presents that LSTM needs to use the appropriate number of layers for training. A better model is not necessary engaging in more layers.
D.
Type 4 bidirectional multi-layers
According to the results of multiple layers for the unidirectional LSTM architecture as shown in to Figure 7, the accuracy using 2 layers of LSTM is higher than using 1 layer. Since the correlation on positions between gaze points is close, the accuracy of 3 layers and 4 layers shows a downward trend. The reason why bi-directional is higher than unidirectional LSTM is that for objects of interest, gaze points usually concentrate in a small area in a short time, as shown in Figure 3. Therefore, feedback information is useful to improve the accuracy. When the number of LSTM layers increases, the accuracy shows a downward trend. The possible reason is that if there are not many features, the architecture is too complicated and affects the interpretation. As it can be seen from Figure 7, 1 layer with bidirectional and 2 layers with unidirectional architectures have relatively high accuracy outcomes.
The computational complexity of the network can be estimated by the total parameters of the network. The total parameters for different layers, and unidirectional and bidirectional of LSTM shown in Figure 8 are presented to contrast the efficiency, the more parameters the more layers of LSTM. By observing the results in Figure 7 and Figure 8, the 2-layer unidirectional LSTM has 768,000 more parameters than the 1-layer bidirectional LSTM, and the accuracy rate is also reduced by 0.4%. Therefore, the performance of the 1-layer bidirectional LSTM is used to compare with others networks [34,35].

3.1.2. GAP

This subsection discusses the comparison of the results of using GAP after the LSTM. The GAP can reduce the number of parameters of the model, thus the training time can be reduced accordingly. The experimental results are shown in Table 6. The addition of GAP reduces the overall parameter amount by about 13%, and the accuracy of the model is increased by about 1.2%. In addition, it has a relatively good performance in other evaluation metrics including loss, F1 score, recall and precision.

3.1.3. Dropout and BN

The use of dropout can prevent the occurrence of overfitting and improve the generalization ability of the model. The use of BN can increase the convergence speed during learning, thereby increasing the stability of the neural network. For the evaluation of the influence of making use of dropout and BN on system performance, the overall experimental results are shown in Table 7. The results show that the accuracy 93.1% of the 1- layer LSTM with bidirectional architecture using GAP and BN is the best model. In Table 7, it shows that the performance is improved by adopting the BN method comparing to Table 6.

3.2. The Comparison of Various Models

The result of Section 3.1 is based on the model of CNN in the proposed CCLN using [15]. In order to consider the utility of using the general network architecture, this study evaluates several general CNN models, including VGG16, VGG19, ResNet50, DenseNet121 and MobileNet [34,35]. Evaluating the impact of transfer learning on overall performance is as following.

3.2.1. Comparison of Various General Models

To compare the five commonly used general models with pre-trained ImageNet database and a 1-layer unidirectional LSTM, the experimental results are shown in Table 8. The results show that the accuracy of 71% of the MobileNet model is the highest. In general, the four models of VGG16, VGG19, ResNet50, and DenseNet121, all increase the depth of the network by increasing the number of convolutional layers, thereby increasing the accuracy. This deeper network model is mainly used to identify 1000 categories of ImageNet; however, there are a few object features to be identified in this study since the facial images are used as the input data.
Compared with the previous four deep CNN models as shown in Table 3, the MobileNet model has a lower accuracy by using Pre-trained ImageNet which has few convolutional layers. CCLN uses the facial images as the input data, and the features of facial images are not as complex as those used in ImageNet. Therefore, an accuracy rate of using MobileNet model with Pre-trained ImageNet is 71%, a reduction of about 77% of the total parameters can be achieved.

3.2.2. Comparison of Multi-Layers LSTM

According to Table 8, although the accuracy of MobileNet and DenseNet121 is not much different, the MobileNet has the lowest total parameter amount. The performances of MobileNet cascaded multi-layers LSTM with unidirectional or bidirectional are compared as shown in Table 9 and Table 10. Both of them have the best accuracy with 1 layer of LSTM, and the accuracy of bidirectional has increased to 91.4%, that is, it is better than unidirectional, and the result is consistent with Figure 7. From this experiment, we know that using more layers of LSTM will not make the accuracy higher whether it uses unidirectional or bidirectional. In addition, through the feedback and training of the previous and future feature sequences by the bidirectional LSTM, a better classification output can be made.

3.2.3. GAP

The evaluated results of MobileNet cascaded 1-layer LSTM with bidirectional and using GAP instead of fully connected are shown in Table 11. According to the results, using GAP, the total parameter amount is reduced by about 7%, the time required for model training is shortened. In addition, the model accuracy rate is improved by 1.2%. Additionally, the performance is improved in other evaluation metrics including loss, F1 score, recall and precision.

3.2.4. Dropout and BN

Using dropout and BN to evaluate MobileNet cascaded 1-layer LSTM with bidirectional and adopting GAP, the experimental results are shown in Table 12. Even if the accuracy of the model is not much different from the unused one, both are around 92%. It is known that dropout can improve the generalization ability and using BN can speed up the convergence. Therefore, the impacts of using dropout and BN to MobileNet on the performance are also evaluated. Among them, the accuracy, F1-score, recall and precision of cascading 1-layer LSTM with bidirectional and using GAP are the highest. Using dropout will cause the same results as shown in Table 7 causing a performance degradation.

3.2.5. Transfer Learning

The previous experiment using general model is based on ImageNet data for pre-training. The content of ImageNet is a collection of general objects, including people features. According to the model proposed in [15], gaze estimation is trained with faces as the input data, that is, facial image input is used for pre-training to extract features. Therefore, this paper also evaluates the effect of adding face and eyes to pre-train on the performance by using the images acquired from [15], both with 8000 images, respectively. The results are shown in Table 13.
According to the results in Table 13, it is known that whether adding faces or eyes for CNN network pre-training, the accuracy is about 89% to 92%, which is not much different from the best model obtained in the previous experiment, 93% in the Table 7. It can be observed that using the pre-trained model with added images of faces and eyes cannot outperform the best model obtained in the previous experiment. Consequently, the CCLN in this paper proposes to use video as the input data. Whether CNN uses the pre-trained model of the paper [15] or uses the general model of MobileNet as ImageNet to pre-train, the performance can be improved by 8.1% and 7.4%, respectively. The total number of parameters of the best model obtained in the previous experiment as shown in Table 7 is about 9.5% less than the pre-trained models in Table 13.

4. Conclusions

This paper proposes a CCLN architecture with video as input to estimate gaze. By studying the impact of the number layers of LSTM, unidirectional or bidirectional, BN, dropout and GAP on detection accuracy to obtain optimal hyperparameters, we designed a simple, yet effective CCLN. The results show that using the CNN developed in [15] with a cascaded 1-layer LSTM with bidirectional and using GAP and BN has the highest accuracy of 93.1%. In case of adopting general model MobileNet, using ImageNet as the pre-trained model instead of CNN has the best accuracy 92.6%, as shown in Table 12. We also study the impact of transfer learning by adding face and eyes images to pre-train. The results show that it is not helpful to improve the accuracy of the model. It can be observed that whether using a CNN with a specially modified architecture or a general model widely used, the accuracy of the model can be improved for our proposed CCLN.
In the literature, BN is widely used to improve performance. The proposed CCLN of this paper also confirms the improvement results of using BN. Using BN to improve performance also can be found in the model proposed in [15] or in general models of CNN. Nevertheless, the performance of evaluation metrics, including accuracy, loss, F1 score, recall and precision, cannot be improved by adopting dropout.
By inputting facial images with different features such as beards to evaluate the performance will be the future directions. In addition, using GAN (Generative Adversarial Network) to generate other facial features, such as the face costumes of entertainers, and to evaluate the impact introduced by different features, is an attractive issue.

Author Contributions

Conceptualization, B.-J.H.; data curation, B.-J.H., C.-H.H., H.-H.C. and D.-Y.H.; formal analysis, B.-J.H.; methodology, B.-J.H. and C.-H.H.; software, H.-H.C. and D.-Y.H.; validation, B.-J.H. and H.-H.C.; writing—original draft, H.-H.C.; writing—review and editing, B.-J.H. and C.-H.H. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported by the Ministry of Science and Technology (MOST), Taiwan, Republic of China, under Contract of MOST 109-2410-H-130-044.

Data Availability Statement

If researchers are interested in using the collected datasets, the datasets can be made available by sending request emails to the authors.

Conflicts of Interest

The authors declare no conflict of interest.

References

  1. Tang, D.L.; Lee, T.R.; Tsai, C.M. An Exploratory Study on Relationship between Preference and Scanpath-Evidence from Color Preference Sorting Task. Chin. J. Psychol. 2005, 47, 339–351. [Google Scholar]
  2. Anuradha, A.; Corcoran, P. A review and analysis of eye-gaze estimation systems, algorithms and performance evaluation methods in consumer platforms. IEEE Access 2017, 5, 16495–16519. [Google Scholar]
  3. John, K.K.; Jensen, J.D.; King, A.J.; Pokharel, M.; Grossman, D. Emerging applications of eye-tracking technology in dermatology. J. Dermatol. Sci. 2018, 91, 117–123. [Google Scholar] [CrossRef] [PubMed] [Green Version]
  4. Hoppe, S.; Bulling, A. End-to-end eye movement detection using convolutional neural networks. arXiv 2016, arXiv:1609.02452. [Google Scholar]
  5. Zhu, W.; Deng, H. Monocular free-head 3d gaze tracking with deep learning and geometry constraints. In Proceedings of the IEEE International Conference on Computer Vision (ICCV), Venice, Italy, 2–29 October 2017; pp. 3162–3171. [Google Scholar]
  6. Lemley, J.; Kar, A.; Drimbarean, A.; Corcoran, P. Convolutional Neural Network Implementation for Eye-Gaze Estimation on Low-Quality Consumer Imaging Systems. IEEE Trans. Consum. Electron. 2019, 65, 179–187. [Google Scholar] [CrossRef] [Green Version]
  7. Chen, H.H.; Hwang, B.J.; Hsu, W.H.; Kao, C.W.; Chen, W.T. Focus of Area Tracking Based on Deep Learning for Multiple Users. IEEE Access 2019, 7, 179477–179491. [Google Scholar] [CrossRef]
  8. Zhang, X.; Sugano, Y.; Fritz, M.; Bulling, A. Appearance-based gaze estimation in the wild. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Boston, MA, USA, 7–12 June 2015; pp. 4511–4520. [Google Scholar]
  9. Sugano, Y.; Matsushita, Y.; Sato, Y. Learning-by-synthesis for appearance-based 3d gaze estimation. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Columbus, OH, USA, 24–27 June 2014; pp. 1821–1828. [Google Scholar]
  10. Krafka, K.; Khosla, A.; Kellnhofer, P.; Kannan, H.; Bhandarkar, S.; Matusik, W.; Torralba, A. Eye tracking for everyone. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Las Vegas, NV, USA, 26 June–1 July 2016; pp. 2176–2184. [Google Scholar]
  11. Funes Mora, K.A.; Monay, F.; Odobez, J.M. Eyediap: A database for the development and evaluation of gaze estimation algorithms from rgb and rgb-d cameras. In Proceedings of the ACM Symposium on Eye Tracking Research and Applications, Safety Harbor, FL, USA, 26–28 March 2014; pp. 255–258. [Google Scholar]
  12. McMurrough, C.D.; Metsis, V.; Kosmopoulos, D.; Maglogiannis, I.; Makedon, F. A dataset for point of gaze detection using head poses and eye images. J. Multimodal User Interfaces 2013, 7, 207–215. [Google Scholar] [CrossRef]
  13. Huang, Q.; Veeraraghavan, A.; Sabharwal, A. TabletGaze: Dataset and analysis for unconstrained appearance-based gaze estimation in mobile tablets. Mach. Vis. Appl. 2017, 28, 445–461. [Google Scholar] [CrossRef]
  14. Cerrolaza, J.J.; Villanueva, A.; Cabeza, R. Study of Polynomial Mapping Functions in Video-Oculography Eye Trackers. ACM Trans. Comput.-Hum. Interact. 2012, 19, 1–25. [Google Scholar] [CrossRef]
  15. Chen, H.H.; Hwang, B.J.; Wu, J.S.; Liu, P.T. The Effect of Different Deep Network Architectures upon CNN-Based Gaze Tracking. Algorithms 2020, 13, 127. [Google Scholar] [CrossRef]
  16. Hochreiter, S.; Schmidhuber, J. Long short-term memory. Neural Comput. 1997, 9, 1735–1780. [Google Scholar] [CrossRef]
  17. Olah, C. Understanding LSTM Networks. 2015. Available online: http://colah.github.io/posts/2015-08-Understanding-LSTMs/ (accessed on 10 January 2021).
  18. Chen, K.; Zhou, Y.; Dai, F. A LSTM-based method for stock returns prediction: A case study of China stock market. In Proceedings of the IEEE International Conference on Big Data (Big Data), Santa Clara, CA, USA, 29 October–1 November 2015; pp. 2823–2824. [Google Scholar]
  19. Ullah, A.; Ahmad, J.; Muhammad, K.; Sajjad, M.; Baik, S.W. Action recognition in video sequences using deep Bi-directional LSTM with CNN features. IEEE Access 2017, 6, 1155–1166. [Google Scholar] [CrossRef]
  20. Zhang, Z.; Lv, Z.; Gan, C.; Zhu, Q. Human action recognition using convolutional LSTM and fully-connected LSTM with different attentions. Neurocomputing 2020, 410, 304–316. [Google Scholar] [CrossRef]
  21. Liu, Z.; Li, Z.; Wang, R.; Zong, M.; Ji, W. Spatiotemporal saliency-based multi-stream networks with attention-aware LSTM for action recognition. Neural Comput. Appl. 2020, 32, 14593–14602. [Google Scholar] [CrossRef]
  22. Majd, M.; Safabakhsh, R. Correlational Convolutional LSTM for human action recognition. Neurocomputing 2020, 396, 224–229. [Google Scholar] [CrossRef]
  23. Husain, S.S.; Natarajan, B. Enhanced CNN-LSTM approach for Human Activity Recognition. Int. J. Mod. Agric. 2020, 9, 199–205. [Google Scholar]
  24. Zhao, Y.; Man, K.L.; Smith, J.; Siddique, K.; Guan, S.U. Improved two-stream model for human action recognition. EURASIP J. Image Video Process. 2020, 2020, 1–9. [Google Scholar] [CrossRef]
  25. Koochaki, F.; Najafizadeh, L. Eye gaze-based early intent prediction utilizing cnn-lstm. In Proceedings of the 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC), Berlin, Germany, 23–27 July 2019; pp. 1310–1313. [Google Scholar]
  26. Lin, M.; Chen, Q.; Yan, S. Network in network. arXiv 2013, arXiv:1312.4400. [Google Scholar]
  27. Ioffe, S.; Szegedy, C. Batch normalization: Accelerating deep network training by reducing internal covariate shift. arXiv 2015, arXiv:1502.03167. [Google Scholar]
  28. Xu, J.; Jiang, M.; Wang, S.; Kankanhalli, M.S.; Zhao, Q. Predicting human gaze beyond pixels. J. Vis. 2014, 14, 1–20. [Google Scholar] [CrossRef] [Green Version]
  29. Yun, K.; Peng, Y.; Samaras, D.; Zelinsky, G.J.; Berg, T.L. Studying relationships between human gaze, description, and computer vision. In Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Portland, OR, USA, 25–27 June 2013; pp. 739–746. [Google Scholar]
  30. Zhang, H.; Zhang, Q.; Shao, S.; Niu, T.; Yang, X. Attention-based LSTM network for rotatory machine remaining useful life prediction. IEEE Access 2020, 8, 132188–132199. [Google Scholar] [CrossRef]
  31. Xingjian, S.H.I.; Chen, Z.; Wang, H.; Yeung, D.Y.; Wong, W.K.; Woo, W.C. Convolutional LSTM network: A machine learning approach for precipitation nowcasting. In Advances in Neural Information Processing Systems; Curran Associates: Red Hook, NY, USA, 2015; pp. 802–810. [Google Scholar]
  32. Chen, Y.; Li, W.; Wang, L.; Hu, J.; Ye, M. Vision-based fall event detection in complex background using attention guided bi-directional LSTM. IEEE Access 2020, 8, 161337–161348. [Google Scholar] [CrossRef]
  33. Ullah, W.; Ullah, A.; Haq, I.U.; Muhammad, K.; Sajjad, M.; Baik, S.W. CNN features with bi-directional LSTM for real-time anomaly detection in surveillance networks. Multimed. Tools Appl. 2021, 80, 16979–16995. [Google Scholar] [CrossRef]
  34. Umair, M.; Khan, M.S.; Ahmed, F.; Baothman, F.; Alqahtani, F.; Alian, M.; Ahmad, J. Detection of COVID-19 Using Transfer Learning and Grad-CAM Visualization on Indigenously Collected X-ray Dataset. Sensors 2021, 21, 5813. [Google Scholar] [CrossRef] [PubMed]
  35. Pan, L.; Pipitsunthonsan, P.; Daengngam, C.; Channumsin, S.; Sreesawe, S.; Chongcheawchamnan, M. Identification of complex mixtures for Raman spectroscopy using a novel scheme based on a new multi-label deep neural network. IEEE Sens. J. 2021, 21, 10834–10843. [Google Scholar] [CrossRef]
  36. Howard, A.G.; Zhu, M.; Chen, B.; Kalenichenko, D.; Wang, W.; Weyand, T.; Andreetto, M.; Adam, H. Mobilenets: Efficient convolutional neural networks for mobile vision applications. arXiv 2017, arXiv:1704.04861. [Google Scholar]
Figure 1. System architecture.
Figure 1. System architecture.
Sensors 22 00545 g001
Figure 2. Most gaze points are distributed near the center of the interested object [28].
Figure 2. Most gaze points are distributed near the center of the interested object [28].
Sensors 22 00545 g002
Figure 3. Comparing the gaze points of the different participant during frame 2387 to frame 2598.
Figure 3. Comparing the gaze points of the different participant during frame 2387 to frame 2598.
Sensors 22 00545 g003
Figure 4. The distribution results after labeled as blocks.
Figure 4. The distribution results after labeled as blocks.
Sensors 22 00545 g004
Figure 5. The distribution results after data augmentation.
Figure 5. The distribution results after data augmentation.
Sensors 22 00545 g005
Figure 6. The architecture of CCLN Gaze Estimation Network.
Figure 6. The architecture of CCLN Gaze Estimation Network.
Sensors 22 00545 g006
Figure 7. The accuracy of multiple layers for different LSTM architecture.
Figure 7. The accuracy of multiple layers for different LSTM architecture.
Sensors 22 00545 g007
Figure 8. The total parameters for different layers, and unidirectional and bidirectional of LSTM.
Figure 8. The total parameters for different layers, and unidirectional and bidirectional of LSTM.
Sensors 22 00545 g008
Table 1. The details of six animated videos [15].
Table 1. The details of six animated videos [15].
TitleResolution (Pixel)Length (s)
Reach1920 × 1080193
Mr Indifferent1920 × 1080132
Jinxy Jenkins & Lucky Lou1920 × 1080193
Changing Batteries1920 × 1080333
Pollo1920 × 1080286
Pip1920 × 1080245
Table 2. The number of video lengths.
Table 2. The number of video lengths.
Length (s)Amount
1991
221
35
Table 3. The types of LSTM Block architectures to be evaluated.
Table 3. The types of LSTM Block architectures to be evaluated.
TypeCNNLSTM
1CNN Blockunidirectional
2CNN Blockbidirectional
3CNN Blockunidirectional multi-layers
4CNN Blockbidirectional multi-layers
Table 4. The accuracy compared with [15] for Type 1.
Table 4. The accuracy compared with [15] for Type 1.
[15]Proposed CCLN
Accuracy0.8620.891
Table 5. The accuracies for Type 1 and Type 2.
Table 5. The accuracies for Type 1 and Type 2.
Type 1Type 2
Accuracy0.8910.915
Table 6. The performance for using GAP.
Table 6. The performance for using GAP.
1-Layer Bidirectional LSTM1-Layer Bidirectional LSTM + GAP
Total parameters6,345,1575,839,301
Accuracy0.9150.927
Loss0.3610.324
F1 score0.9150.929
Recall0.9190.935
Precision0.9160.929
Table 7. The performance for using Dropout and BN.
Table 7. The performance for using Dropout and BN.
1-Layer Bidirectional LSTM + GAP + Dropout1-Layer Bidirectional LSTM + GAP + BN1-Layer Bidirectional LSTM + GAP + Dropout + BN
Total parameters5,839,3015,841,3495,841,349
Accuracy0.9180.9310.926
Loss0.3020.3590.351
F1 score0.9190.9300.925
Recall0.9230.9350.931
Precision0.9200.9300.926
Table 8. The performance for various general models.
Table 8. The performance for various general models.
CNN ModelVGG16VGG19ResNet50DenseNet121MobileNet
Pre-trained ImageNet ImageNet ImageNet ImageNet ImageNet
Total parameters19,194,72524,504,42129,061,15727,294,9176,636,389
Accuracy 0.60.6160.6850.7050.71
Loss 1.51.381.181.061
F1 score 0.5880.5960.6710.6930.7
Recall 0.6050.6210.6920.7090.718
Precision 0.5970.6010.6640.6880.7
Table 9. The performance of MobileNet [36] cascaded multi-layers LSTM with unidirectional.
Table 9. The performance of MobileNet [36] cascaded multi-layers LSTM with unidirectional.
LSTM Layer1 Layer2 Layers3 Layers
Pre-trainedImageNetImageNetImagenet
Total parameters6,636,3898,735,58910,834,789
Accuracy0.710.6250.572
Loss11.271.45
F1 score0.70.6070.54
Recall0.7180.6340.578
Precision0.70.6180.537
Table 10. The performance of MobileNet [36] cascaded multi-layers LSTM with bidirectional.
Table 10. The performance of MobileNet [36] cascaded multi-layers LSTM with bidirectional.
LSTM Layer1 Layer2 Layers3 Layers
Pre-trainedImageNetImageNetImageNet
Total parameters6,900,58113,196,13319,491,685
Accuracy0.9140.8760.863
Loss0.3580.5490.71
F1 score0.9130.8790.865
Recall0.920.8780.868
Precision0.9140.8890.87
Table 11. The performance of MobileNet with or without adopting GAP.
Table 11. The performance of MobileNet with or without adopting GAP.
GAPWithoutWith
Pre-trainedImageNetImageNet
Total parameters6,900,5816,394,725
Accuracy0.9140.926
Loss0.3580.317
F1 score0.9130.926
Recall0.920.929
Precision0.9140.928
Table 12. The performance of MobileNet using dropout or BN.
Table 12. The performance of MobileNet using dropout or BN.
WithoutDropoutBNDropout + BN
Pre-trainedImageNetImageNetImageNetImageNet
Total parameters6,394,7256,394,7256,396,7736,396,773
Accuracy0.9260.920.9230.922
Loss0.3170.3050.4070.287
F1 score0.9260.920.9240.921
Recall0.9290.9240.9290.925
Precision0.9280.9230.9250.924
Table 13. The performance of transfer learning.
Table 13. The performance of transfer learning.
Pre-TrainedAdd Face and EyesAdd FaceAdd Eyes
Total parameters6,394,7256,394,7256,394,725
Accuracy0.9190.9240.891
Loss0.3410.3320.537
F1 score0.9150.9220.813
Recall0.9180.9210.875
Precision0.9110.9240.883
Publisher’s Note: MDPI stays neutral with regard to jurisdictional claims in published maps and institutional affiliations.

Share and Cite

MDPI and ACS Style

Hwang, B.-J.; Chen, H.-H.; Hsieh, C.-H.; Huang, D.-Y. Gaze Tracking Based on Concatenating Spatial-Temporal Features. Sensors 2022, 22, 545. https://doi.org/10.3390/s22020545

AMA Style

Hwang B-J, Chen H-H, Hsieh C-H, Huang D-Y. Gaze Tracking Based on Concatenating Spatial-Temporal Features. Sensors. 2022; 22(2):545. https://doi.org/10.3390/s22020545

Chicago/Turabian Style

Hwang, Bor-Jiunn, Hui-Hui Chen, Chaur-Heh Hsieh, and Deng-Yu Huang. 2022. "Gaze Tracking Based on Concatenating Spatial-Temporal Features" Sensors 22, no. 2: 545. https://doi.org/10.3390/s22020545

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop