Title: BETANAS: BALANCED TRAINING AND SELECTIVE DROP FOR NEURAL ARCHITECTURE SEARCH

Abstract: Automatic neural architecture search techniques are becoming increasingly important in machine learning area. Especially, weight sharing methods have shown remarkable potentials on searching good network architectures with few computational resources. However, existing weight sharing methods mainly suffer limitations on searching strategies: these methods either uniformly train all network paths to convergence which introduces conflicts between branches and wastes a large amount of computation on unpromising candidates, or selectively train branches with different frequency which leads to unfair evaluation and comparison among paths. To address these issues, we propose a novel neural architecture search method with balanced training strategy to ensure fair comparisons and a selective drop mechanism to reduce conflicts among candidate paths. The experimental results show that our proposed method can achieve a leading performance of 79.0% on ImageNet under mobile settings, which outperforms other state-of-the-art methods in both accuracy and efficiency.

1. INTRODUCTION:

The fast developing of artificial intelligence has raised the demand to design powerful neural networks. Automatic neural architecture search methods (Zoph & Le, 2016; Zhong et al., 2018; Pham et al., 2018) have shown great effectiveness in recent years. Among them, methods based on weight sharing (Pham et al., 2018; Liu et al., 2018; Cai et al., 2018; Guo et al., 2019) show great potentials on searching architectures with limited computational resources. These methods are divided into 2 categories: alternatively training ones (Pham et al., 2018; Liu et al., 2018; Cai et al., 2018) and oneshot based ones (Brock et al., 2017; Bender et al., 2018) . As shown in Fig 2 , both categories construct a super-net to reduce computational complexity. Methods in the first category parameterize the structure of architectures with trainable parameters and alternatively optimize architecture parameters and network parameters. In contrast, one-shot based methods train network parameters to convergence beforehand and then select architectures with fixed parameters. Both categories achieve better performance with significant efficiency improvement than direct search.

Despite of these remarkable achievements, methods in both categories are limited in their searching strategies. In alternatively training methods, network parameters in different branches are applied with different training frequency or updating strength according to searching strategies, which makes different sub-network convergent to different extent. Therefore the performance of sub-networks extracted from super-net can not reflect the actual ability of that trained independently without weight sharing. Moreover, some paths might achieve better performance at early steps while perform not well when actually trained to convergence. In alternatively training methods, these operators will get more training opportunity than other candidates at early steps due to their well performance. Sufficient training in turn makes them perform better and further obtain more training opportunities, forming the Matthew Effect. In contrast, other candidates will be always trained insufficiently and can never show their real ability.

Differently, One-shot methods train paths with roughly equal frequency or strength to avoid the Matthew Effect between parameters training and architectures selection. However, training all paths to convergence costs multiple time. Besides, the operators are shared by plenty of sub-networks, making the backward gradients from different training steps heavily conflict. To address this issue, Experiments are conducted on ImageNet classification task. The searching process costs less computational resources than competing methods and our searched architecture achieves an outstanding accuracy of 79.0%, which outperforms state-of-the-art methods under mobile settings. The proposed method is compared with other competing algorithms with visualized analysis, which demonstrates its the effectiveness. Moreover, we also conduct experiments to analysis the mutual interference in weight sharing and demonstrate the rationality of the gradually drop paths strategy.

2. RELATED WORK:

Automatic neural architecture search techniques has attracted much attention in recent years. NAS-Net (Zoph & Le, 2016; Zoph et al., 2018) proposes a framework to search for architectures with reinforcement learning, and evaluates each of the searched architectures by training it from scratch. BlockQNN (Zhong et al., 2018; Guo et al., 2018; Zhong et al., 2018) expands the search space to the entire DAG and selects nets with Q-learning. Network pruning methods (Li et al., 2019; Noy et al., 2019) prune redundant architectures to reduces search spaces. Considering the searching policy, most of these methods depend on reinforcement learning, evolutionary algorithms and gradient based algorithms (Bello et al., 2017; Liu et al., 2018; Cai et al., 2018) .

The most related works to our method are the ones based on weight sharing proposed by (Pham et al., 2018) , from which two streams are derived: Alternatively training methods (Cai et al., 2018; Liu et al., 2018) and one-shot methods (Brock et al., 2017; Bender et al., 2018; Guo et al., 2019) . Methods in the first stream alternatively train architecture parameters and network parameters. During search process, operators in the super-net are selectively trained and evaluated with a certain policy and the policy is updated dynamically according to the evaluations. Among them, ENAS (Pham et al., 2018) introduces RL to select paths. DARTS (Liu et al., 2018) improves the accuracy and efficiency of paths selection policy by considering the importance of each path as trainable parameters. ProxyLessNAS (Cai et al., 2018) proposes to directly search on target datasets with single paths and makes the latency term differentiable. Single-Path NAS (Stamoulis et al., 2019) 2019) uses mask to select paths. (Sciuto et al., 2019) Our work benefits from the advantages of both categories: On one hand, the importance factors are evaluated with a gradient based approach but it has no influence on training shared parameters. On the other hand, the shared parameters are updated uniformly as those in one-shot.

3. APPROACH:

ProxyLessNAS (Cai et al., 2018) and Single Path One-shot (Guo et al., 2019) proposed to train the super-net with only one path on in each step to make the performance trained with weight sharing more close to that trained alone. Both of them enhance the performance of weight sharing to a higher level. ProxyLessNAS updates architecture parameters and network parameters alternatively.

Paths are selectively trained according to their performance, and paths with higher performance get more training opportunities. Single Path One-shot first proposed to balanced train all paths until convergence and then use evolution algorithms to select network structures. The equivalent functions of the choice blocks in two methods are described as m P L and m OS in Eq 1:

EQUATION

Our method follows the alternatively training ones, in which architecture parameters and network parameters are optimized alternatively in each step. To give a better solution to the problems discussed above, we train each candidate path with equal frequency to avoid the "Matthew effect" and gradually dropped least promising paths during searching process to reduce conflicts among candidate paths.

3.1. PIPELINE:

The pipeline of our method is shown in Algorithm 1. First of all, a super-net is constructed with

L choice blocks O 1 , O 2 , . . . , O L , as shown in Fig 1. Each choice block O l is composed of M candidate paths and corresponding operators o l,1 , o l,2 , . . . , o l,M .

The importance factor of o l,m is denoted as α l,m and α l,m are converted to probability factor p l,m using softmax normalization.

Secondly, the parameters of o l,m and their importance factors α l,m are trained alternatively in Phase 1 and Phase 2. When training α l,m , latency term is introduced to balance accuracy and complexity. Paths with α l,m lower than th α will be dropped and no more trained.

Algorithm 1 Searching Process Initialization: Denote O l as the choice block for layer l with M candidate operators {o l,1 , o l,2 , . . . , o l,M }. α l,1 , α l,2 , . . . , α l,M are the corresponding importance factors of candidate operators and initialized with identical value. S max denotes the max number of optimization steps.

1: while t < S max do 2:

Phase1: Randomly select o l,m l ∈ O l for block O l with uniform probability, then fix all α l,m and train the super-net constructed with the selected o 1,m1 , o 2,m2 , . . . , o L,m L for some steps.

3::

Phase2: Fix all the parameters in o l,m and measure their flops/latency. Then evaluate each operator o l,m with both cross-entropy loss and flops/latency loss. Update α l,m according to the losses feedback.

4:

for o l,m ∈ O l do 5: if α l,m < th α then O l = O l \ {o l,m } t = t + 1 6: for o l,m ∈ O l do m l = argmax m (α l,m ) 7: return o 1,m1 , o 2,m2 , . . . , o L,m L

Finally, after alternatively training o l,m and α l,m for given steps, paths with the highest importance factor in each choice block are selected to compose a neural architecture as the searching result.

3.2. BALANCED TRAINING:

Alternatively training methods focus computational resources on most promising candidates to reduce the interference from redundant branches. However, some operators that perform well at early phases might not perform as well when they are trained to convergence. These operators might get much more training opportunities than others due to their better performance at the beginning steps. Higher training frequency in turn maintains their dominant position in the following searching process regardless their actual ability, forming the Matthew Effect. In contrast, the operators with high performance when convergent might never get opportunities to trained sufficiently. Therefore, the accuracy of alternatively training methods might degrade due to inaccurate evaluations and comparison among candidate operators.

Our method follows the alternatively optimizing strategy. Differently, we only adopt gradient to architectures optimization while randomly sample paths with uniformly probability when training network parameters to avoid the Matthew Effect. More specifically, when updating network parameters of o l,m in Phase 1 and architecture parameters in Phase 2, the equivalent output of choice block O l is given as O path l in Eq 2 and O arch l in Eq 3:

O path l (x) = o l,m (x)

with probability 1 M , if α l,m > th α with probability 0 , else.

(2)

EQUATION

)

Where M is the number of remaining operators in O l currently, and p l,m is the softmax form of α l,m . The α l,m of dropped paths are not taken into account when calculating p l,m . The parameters in both phases are optimized with Stochastic Gradient Descent (SGD). In Phase 1, the outputs in Eq 2 only depends on network parameters, thus gradients can be calculated with the Chain Rule. In Phase 2, the outputs not only depend on the fixed network parameters but also architecture parameters α l,m . Note that O arch l (x) is not differentiable with respect to α l,m , thus we introduce the manually defined derivatives proposed by Cai et al. (2018) to deal with this issue: Eq 3 can be expressed as

O arch l (x) = g l,m • o l,m (x)

, where g l,0 , g l,0 , . . . , g l,M is a one-hot vector with only one element equals to 1 while others equal to 0. Assuming ∂g l,j /∂p l,j ≈ 1 according to Cai et al. (2018) , the derivatives of O arch l (x) w.r.t. α l,m are defined as :

EQUATION

From now on, O path l (x) and O arch l (x) are differentiable w.r.t. network parameters and architecture parameters respectively. Both parameters can be optimized alternatively in Phase 1 and Phase 2.

3.3. SELECTIVELY DROP PATHS:

One-shot based methods, such as Single Path One-shot Guo et al. (2019) also uniformly train paths. These methods train network parameters of each path uniformly to convergence, after that a searching policy is applied to explore a best structure with fixed network parameters. However, the optimizations of candidate operators in a same choice block actually conflict. Considering N candidate operators in a same choice block and their equivalent functions f 1 , f 2 , . . . , f N , given F in and F out as the equivalent functions of the sub-supernet before and after the current choice block, x i and y i as input data and labels from the training dataset, and L as the loss metric, the optimization of network parameters can be described as:

EQUATION

When the super-net is trained to convergence, F in and F out is comparatively stable. In this situation, In contrast, each operator process data from only one input when networks are trained independently.

f 1 (w 1 ), f 2 (w 2 ), . . . , f N (w N )

Both problems widen the gap between network trained with and without weight sharing.

Fewer candidate paths help reduce the conflicts among operators, which will explained in the experiments section. Therefore, a drop paths strategy is applied in our method to reduce mutual interference among candidate operators during searching process. The paths with performance lower than a threshold will be permanently dropped to reduce its influence on remaining candidates.

When updating α l,m in phase 2, we follow the strategy in ProxyLessNAS (Cai et al., 2018) to sample path in each choice block with probability p l,m , and optimize α l,m by minimizing the expectation joint loss L:

EQUATION

where L CE and L LA are the expectation cross-entropy loss and latency loss, L LA l,m is the flops or latency of operator o l,m and β is a hyper-parameter to balance two loss terms. We regard the importance factor α l,m and its softmax form p l,m as sampling probability and use Eq 6 to optimize α l,m . The derivatives of L CE and L LA w.r.t. α l,m can be get from Eq 4 and 6 respectively. Note that α l,m is only applied to evaluate the importance of paths and have no influence on the balanced training strategy in Phase 1. After each step of evaluation, paths with low α l,m are dropped and will not be trained anymore:

O l = O l \ {o l,m l , if α l,m l < th α , ∀m l } (7)

The limitations of alternatively training methods and one-shot based ones are relieved by the proposed balanced training and drop paths strategies respectively. The two phases are trained alternatively until meeting the stop condition, then paths with highest α l,m from each block are selected to compose an architecture as the searching result.

4. EXPERIMENTS:



4.1. EXPERIMENTAL SETTINGS:

Datasets The target architecture is directly searched on ImageNet (Deng et al., 2009) for classification task. 50000 images are extracted from training set to train architecture parameters α l,m , and the rest of training dataset is used to train network weights.

Search Space We follow the search space in MNASNet (Tan et al., 2018) , where each choice block includes an identity operator and 6 MobileNetV2 blocks which have kernel size 3,5,7 and expand ratio 3, 6 respectively. SE-Layer (Hu et al., 2018) is not applied to the search space.

Training Detail We search on V100 GPUs for 160 GPU hours. The shared parameters are trained with 1024 batch size and 0.1 learning rate. α l,m is trained with Adam optimizer and 1e-3 initial learning rate. Finally, the searched architecture is trained from scratch according to the setting of MobileNetV2 (Sandler et al., 2018) . The searched networks are trained from scratch on training dataset with hyper-parameters as follows: batch size 2048, learning rate 1.4, weight decay 2e-5, cosine learing rate for 350 epochs, dropout 0.2, label smoothing rate 0.1. 

4.2. EXPERIMENTAL RESULTS:

Table 1 : BetaNet-A compared with the state-of-the-art methods under comparable flops.

Method Top1 Top5 Flops params cost (%) (%) (M) (M) (GPU*h)

MobileNet V2 (Sandler et al., 2018) 72.0 91.0 300 3.4 -DARTS (Liu et al., 2018) 73.1 91.0 595 4.9 96 FBNet-C (Wu et al., 2018) 74.9 -375 --ProxyLessNAS (Cai et al., 2018) 74.6 92.2 320 -200 MNASNet+SE (Tan et al., 2018) 75.4 92.5 317 4.2 40,000 MobileV3+SE+swish (Howard et al., 2019) 75 MobileNet V2 (Sandler et al., 2018) 72.0 91.0 6.1 -ShuffleNetV2 (1.5) (Ma et al., 2018) 72.6 -7.3 -ProxyLessNAS-gpu (Cai et al., 2018) achieve the best performance. SENet (Hu et al., 2018) with ratio 0.0625 is applied in table 1 as BetaNet-A + SE. BetaNet-A performs better than MobileNetV2 with comparable flops by 3.1%.

BetaNet-B performs better with comparable latency by 3.8%.Auto-augment (Cubuk et al., 2018) and SWISH activation (Ramachandran et al., 2017) are also applied to the searched BetaNet-A and the performance is further enhanced to 79.0%. As shown in Fig 4 , BetaNet-A outperforms MobileNetV2, ProxyLessNAS (Cai et al., 2018) and MNASNet (Tan et al., 2018) with various depth multiplier.

4.3. ABLATION STUDIES:



4.3.1. COMPARED WITH GRADIENT BASED METHODS:

Experiments are conducted in this sub-section to analyze the contribution of the proposed searching approach: balanced training and selective drop. The searching process of BetaNet-A is compared with that of ProxyLessNAS (Cai et al., 2018) . methods train remaining with much more frequency than ProxyLessNAS, while ProxyLessNAS spend much steps on redundant paths with little promising. 4.3.2 EXPERIMENTS WITH DIFFERENT RANDOM SEEDS (Sciuto et al., 2019) suggests to evaluate weight sharing methods with different random seeds. To examine the robustness of the proposed method, experiments are conducted with our approach with 8 random seeds. 8 architectures are searched independently. We also randomly sample 8 architectures from the same search space and super-net. Constraint is applied to insure both searched and randomly sampled networks have similar flops with BetaNetA. All the 16 networks are trained with the same setting. As shown in Fig 6, all the searched architectures have similar performance on ImageNet. Our method outperforms the random sampled ones and other competing methods with robust performance.

4.3.3. ANALYSIS OF DROP PATH:

Besides the policy of selecting architecture parameters, the optimization of network parameters of our method is different from that of one-shot methods mainly in the drop paths strategy. To give a better understanding of its effectiveness, we conduct experiments on 4 network architectures with 3 different training policies to explore the influence of the number of candidate branches on weight sharing. As shown in Figure7, networks in 3 groups are trained with none weight sharing(NS), 2-branch weight sharing branches(B2) and 4-branch weight sharing branches(B4), respectively. Networks in NS, B2, B4 are trained on cifar10 for 30, 60, 120 epochs respectively to insure network parameters are trained with comparable steps in each group.

The accuracy of the 4 networks trained with NS, B2 and B4 are shown in Fig7 respectively. The experiments indicate 2 phenomenons: 1. The accuracy trained via B2 with 60 epochs is much higher than B4 with 120 epochs, which indicates that less branches in weight sharing helps network parameters converge better. 2. The relative rank of the 4 networks trained with B2 is more similar to those with NS than those with B4, which indicates that less branch can give a better instruction in network selection and demonstrate the rationality of our drop paths strategy.

4.3.4. COMPARISON WITH OTHER WEIGHT SHARING METHODS ON SEARCH STRATEGIES:

Our searching and optimizing strategy is compared to those of other weight sharing approaches, as shown in Tab 3. Alternatively training methods tend to adopt similar strategies to decide which path to train in optimizing phases and which path to evaluate in search phases. However searching strategies are designed to select paths and inevitably treat candidates discriminatively. Therefore, Some of the methods sample a single sub-net in each step while training super-net instead of summing up features from all possible paths to bridge the gap between training super-net and evaluating subnets. Our methods follow this strategy to improve performance and reduce GPU memory consumption as well. To deal with mutual interference among candidates in a same choice block. Progressive-Darts Chen et al. (2019) and our method make effort to drop redundant paths to reduce the interference from them.

4.4. DISCUSSION ON TWO STREAMS OF WEIGHT SHARING METHODS:

There is always a compromise between accuracy and efficiency. Weight sharing shows remarkable improvements in reducing searching cost though introduce inevitable bias on accuracy. Alternatively training approaches are actually greedy methods, since they are talented at searching for next optimal solution. Meanwhile, architectures with less competition at early phases are abandoned. In contrast, one-shot methods attempt offer equal opportunities to all candidates which leads to a more global solution. However operators via weight sharing should deal with outputs from multiple former paths, which is challenging for operators with less flops. Therefore these operators suffer more from the mutual interference than those with larger flops.

Our approach tries to balance the advantages and disadvantages of both streams. On one hand, we tries to insure the accuracy of most promising operators which is similar to the strategies in alternatively training ones. By contrast, only the operators with performance much lower than the average of others will be dropped. On the other hand, we train paths balanced follows the strategies one-shot based ones. Unlikely, paths are gradually dropped to a lower amount in our method to reduce conflicts.

5. CONCLUSION:

This work proposes a novel neural architecture search method via balanced training and selective drop strategies. The proposed methods benefits from both streams of weight sharing approaches and relieve their limitations in optimizing the parameters in super-net. Moreover, our method achieves a new state-of-the-art result of 79.0% on ImageNet under mobile settings with even less searching cost, which demonstrates its effectiveness.

:



Figures and Tables:

Figure 1: Balanced training and selective drop strategy. The evaluation of each candidate only influence architecture search strategy, and paths with low performance will be gradually dropped to reduce conflicts among paths. In addition, paths still remaining are all trained with comparable frequency to insure a fair comparison among candidate operators.Input

are actually trained to fit a same function. However when operators are trained independently without weight sharing, different operators are unlikely to output same feature maps. Take the super-net in Fig 2(b) as an example, the four operators in choice block 1 are likely to be optimized to fit each other. Intuitively, Conv3 and Identity are unlikely to fit a same function when trained without weight sharing. On the other hand, the operators in the second choice block are trained to be compatible with various input features from different operators in the first choice block.

Figure 3: The searched architectures with different complexity limitations: BetaNet-A is searched with flops limitation. BetaNet-B is searched with latency limitaion.

Figure 4: BetaNet-A is compared with MobileNetV2 (Sandler et al., 2018) ProxyLessNAS Cai et al.(2018)  and MNASNet(Tan et al., 2018) with various depth multiplier (0.75, 1.0, 1.4).

Fig 5(a) and (b) show the variation of architecture parameters α and updating frequency of 2 choice blocks for each method respectively. In the result of ProxyLessNAS shown in Fig 5(a) (top left), the conv3_exp6 operator performs better at early phases and get much more training opportunity than other operators and thus trained more sufficiently, which might lead to degraded performance due to the above mentioned "Matthew Effect". Differently, our strategy ensures that all remaining paths are trained with roughly equal frequency. In addition, our

Figure 6: Comparison between architectures searched with different random seeds and randomly sampled ones. Architectures searched with our method and with random policy are shown with solid lines and dotted lines respectively.

Figure 7: 4 Networks are trained in 3 groups with different training policies. 00, 01, 10, 11 represent architectures conv1-conv1, conv1-conv3, conv3-conv1, conv3-conv3 respectively.

directly shares weights via super-kernel. By contrast, one-shot based methods Guo et al. (2019); Brock et al. (2017); Bender et al. (2018) firstly train each path in the super-net with equal frequency to convergence, then all the architectures are selected from the super-net and evaluated with fixed parameters. Darts+ Liang et al. (2019) improves Darts with early stop. Progressive-NAS Chen et al. (2019) gradually increases blocks while searching. HM-NAS Yan et al. (

Table content: <table/>

BetaNet-B compared with the state-of-the-art methods under comparable GPU latency.

Table content: <table><tr><td/><td/><td>.2</td><td>-</td><td>219</td><td>5.4</td><td>-</td></tr><tr><td>BetaNet-A</td><td/><td colspan="2">75.1 92.3</td><td>315</td><td>3.6</td><td>160</td></tr><tr><td>BetaNet-A + SE</td><td/><td colspan="2">75.9 92.8</td><td>333</td><td>4.1</td><td>160</td></tr><tr><td colspan="2">MobileNet V2 (Sandler et al., 2018) × 1.4</td><td colspan="2">74.7 92.5</td><td>585</td><td>6.9</td><td>-</td></tr><tr><td colspan="4">EffNetB0+SE+swish+autoaug Tan &amp; Le (2019) 76.3 93.2</td><td>390</td><td>5.3</td><td>-</td></tr><tr><td colspan="4">EffNetB1+SE+swish+autoaug Tan &amp; Le (2019) 78.8 94.4</td><td>700</td><td>7.8</td><td>-</td></tr><tr><td>BetaNet-A × 1.4</td><td/><td colspan="2">77.1 93.5</td><td>596</td><td>6.2</td><td>160</td></tr><tr><td>MNASNet× 1.4 + SE (Tan et al., 2018)</td><td/><td colspan="2">77.2 93.2</td><td>600</td><td>-</td><td>40000</td></tr><tr><td>BetaNet-A× 1.4 + SE</td><td/><td colspan="2">77.7 93.7</td><td>631</td><td>7.2</td><td>160</td></tr><tr><td colspan="2">BetaNet-A× 1.4 + SE + auto-aug + swish</td><td colspan="2">79.0 94.2</td><td>631</td><td>7.2</td><td>160</td></tr><tr><td>Method</td><td colspan="4">Top1 Top5 GPU Latency</td><td>search cost</td></tr><tr><td/><td>(%)</td><td>(%)</td><td/><td>(ms)</td><td>(GPU hours)</td></tr></table>

Comparison with the other weight sharing methods on their strategies. usually not suitable for optimizing phases. Our method applies different policies in optimizing phases and searching phases to deal with this problem.

Table content: <table><tr><td>Method</td><td>supernet</td><td>search</td><td>sample</td><td>mutual</td></tr><tr><td>ENAS Pham et al. (2018)</td><td>optimization imbalanced</td><td>policy RL</td><td>sub-net √</td><td>interference all candidates</td></tr><tr><td>Darts Liu et al. (2018) SinglePath Guo et al. (2019) ProxyLess Cai et al. (2018)</td><td>imbalanced balanced imbalanced</td><td>gradient EA gradient/RL</td><td>× √ √</td><td>all candidates all candidates all candidates</td></tr><tr><td colspan="2">Progressive Chen et al. (2019) imbalanced</td><td>gradient</td><td>×</td><td>fewer candidates</td></tr><tr><td>Darts+ Liang et al. (2019)</td><td>imbalanced</td><td>gradient</td><td>×</td><td>all candidates</td></tr><tr><td>HM-NAS Yan et al. (2019) Ours</td><td colspan="2">imbalanced gradient+mask balanced gradient</td><td>× √</td><td>all candidates fewer candidates</td></tr><tr><td>searching strategies are</td><td/><td/><td/><td/></tr></table>

Peer review (iclr2020):
In this paper, the authors proposed a new training strategy in achieving better balance between training efficiency and evaluation accuracy with weight sharing-based NAS algorithms. It is consisted of two phrases: in phrase 1, all path are uniformly trained to avoid bias, in phrase 2, less competitive options are pruned to save cost. The proposed method achieved the SOTA on IN mobile setting. Overall I found the idea proposed in the paper intuitive and convincing. Especially I appreciate the ablation study that identified one of the option is encouraged too much in the early stage will can lead to worse final performance. From the methodology perspective, I think this is a solid incremental contribution. However, the highlight of this paper, in my opinion, is the SOTA results on IN. My main concern is that the authors did not indicate that the code/model will be open sourced, which will help verification as well as reproducibility.