Learning One Convolutional Layer with Overlapping Patches

Surbhi Goel, Adam Klivans, Raghu Meka

Introduction

Developing provably efficient algorithms for learning commonly used neural network architectures continues to be a core challenge in machine learning. The underlying difficulty arises from the highly non-convex nature of the optimization problems posed by neural networks. Obtaining provable guarantees for learning even very basic architectures remains open.

The main contribution of this paper is a simple, stochastic update algorithm Convotron (Algorithm 1) for provably learning the above convolutional architecture. The algorithm has the following properties:

Works for general classes of overlapping patches and requires mild distributional conditions.

Proper recovery of the unknown weight vector.

Stochastic in nature with a “gradient-like” update step.

Requires no special/random initialization scheme or tuning of the learning rate.

Tolerates noise and succeeds in the probabilistic concept model of learning.

Logarithmic convergence in 1/ϵ1/\epsilon, the error parameter, in the realizable setting.

This is the first efficient algorithm for learning general classes of overlapping patches (and the first algorithm for any class of patches that succeeds under mild distributional assumptions). Prior work has focused on analyzing SGD in the realizable/noiseless setting with the caveat of requiring either disjoint patches [BG17, DLT+17b] with Gaussian inputs or technical conditions linking the underlying true parameters and the “closeness of patches” [DLT17a].

In contrast, our conditions depend only on the patch structure itself and can be efficiently verified. Commonly used patch structures in computer vision applications such as 1D/2D grids satisfy our conditions. Additionally, we require only that the underlying distribution on samples is symmetric and induces a covariance matrix on the patches with polynomially bounded condition numberBrutzkus and Globerson [BG17] proved that the problem, even with disjoint patches, is NP-hard in general, and so some distributional assumption is needed for efficient learning.. All prior work handles only continuous distributions. Another major difference from prior work is that we give guarantees using purely empirical updates. That is, we do not require an assumption that we have access to exact quantities such as the population gradient of the loss function.

We further show that in the commonly studied setting of Gaussian inputs and non-overlapping patches, updating with respect to a single non-overlapping patch is sufficient to guarantee convergence. This indicates that the Gaussian/no-overlap assumption is quite strong.

2 Our Approach

Our approach is to exploit the monotonicity of the activation function instead of the strong convexity of the loss surface. We use ideas from isotonic regression and extend them in the context of convolutional networks. These ideas have been successful for learning generalized linear models [KKSK11], improperly learning fully connected, depth-three neural networks [GK17b], and learning graphical models [KM17].

3 Related Work

It is known that in the worst case, learning even simple neural networks is computationally intractable. For example, in the non-realizable (agnostic) setting, it is known that learning a single ReLU (even for bounded distributions and unit norm hidden weight vectors) with respect to square-loss is as hard as learning sparse parity with noise [GKKT16], a notoriously difficult problem from computational learning theory. For learning one hidden layer convolutional networks, Brutzkus and Globerson [BG17] proved that distribution-free recoverability of the unknown weight vector is NP-hard, even if we restrict to disjoint patch structures.

As such, a major open question is to discover the mildest assumptions that lead to polynomial-time learnability for simple neural networks. In this paper, we consider the very popular class of convolutional neural networks (for a summary of other recent approaches for learning more general architectures see [GK17a]). For convolutional networks, all prior research has focused on analyzing conditions under which (Stochastic) Gradient Descent converges to the hidden weight vector in polynomial-time.

Along these lines, Brutzkus and Globerson [BG17] proved that with respect to the spherical Gaussian distribution and for disjoint (non-overlapping) patch structures, gradient descent recovers the weight vector in polynomial-time. Zhong et al. [ZSD17] showed that gradient descent combined with tensor methods can recover one hidden layer involving multiple weight vectors but still require a Gaussian distribution and non-overlapping patches. Du et al. [DLT+17b] proved that gradient descent recovers a hidden weight vector involved in a type of two-layer convolutional network under the assumption that the distribution is a spherical Gaussian, the patches are disjoint, and the learner has access to the true population gradient of the loss function.

We specifically highlight the work of Du, Lee, and Tian [DLT17a], who proved that gradient descent recovers a hidden weight vector in a one-layer convolutional network under certain technical conditions that are more general than the Gaussian/no-overlap patch scenario. Their conditions involve a certain “alignment” of the unknown patch structure, the hidden weight vector, and the (continuous) marginal distribution. However, it is unclear which concrete patch-structure/distributional combinations their framework captures. We also note that all of the above results assume there is no noise; i.e., they work in the realizable setting.

Other related works analyzing gradient descent with respect to the Gaussian distribution (but for non-convolutional networks) include [Sol17, GLM17, ZSJ+17, Tia16, LY17, ZPSR17].

In contrast, we consider an alternative to gradient descent, namely Convotron, that is based on isotonic regression. The exploration of alternative algorithms to gradient descent is a feature of our work, as it may lead to new algorithms for learning deeper networks.

Preliminaries

∣∣⋅∣∣||\cdot|| corresponds to the l2l_{2} -norm for vectors and the spectral norm for matrices. The identity matrix is denoted by II. We denote the input-label distribution by D\mathcal{D} over input drawn from X\mathcal{X} and label drawn from Y\mathcal{Y}. The marginal distribution on the input is denoted by DX\mathcal{D}_{\mathcal{X}} and the corresponding probability density function is denoted by PXP_{\mathcal{X}}.

We study the problem of learning the teacher network under the square loss, that is, we wish to find a ww such that

Distribution: The marginal distribution on the input space DX\mathcal{D}_{\mathcal{X}} is a symmetric distribution about the origin, that is, for all xx, PX(x)=PX(−x)P_{\mathcal{X}}(x)=P_{\mathcal{X}}(-x).

Activation Function: The activation function has the following form:

The distributional assumption includes common assumptions such as Gaussian inputs, but is far less restrictive. For example, we do not require the distribution to be continuous nor do we require it to have identity covariance. In Section 4, we show that commonly used patch schemes from computer vision satisfy our patch requirements. The assumption on activation functions is satisfied by popular activations such as ReLU (α=0\alpha=0) and leaky ReLU (α>0\alpha>0).

The activations we consider in this paper have the following useful property:

The loss function can be upper bounded by the l2l_{2}-norm distance of weight vectors using the following lemma.

The Gershgorin Circle Theorem, stated below, is useful for bounding the eigenvalues of matrices.

For a n×nn\times n matrix AA, define Ri:=∑j=1,j≠in∣Ai,j∣R_{i}:=\sum_{j=1,j\neq i}^{n}|A_{i,j}|. Each eigenvalue of AA must lie in at least one of the disks {z:∣z−Ai,i∣≤Ri}\{z:|z-A_{i,i}|\leq R_{i}\}.

Note: The proofs of lemmas in this section have been deferred to the Appendix.

The Convotron Algorithm

In this section we describe our main algorithm Convotron and give a proof of its correctness. Convotron is an iterative algorithm similar in flavor to SGD with a modified (aggressive) gradient update. Unlike SGD (Algorithm 3), Convotron comes with provable guarantees and also does not need a good initialization scheme for convergence.

The following theorem describes the convergence rate of our algorithm:

Define St={(x1,y1),…,(xt,yt)}S_{t}=\{(x_{1},y_{1}),\ldots,(x_{t},y_{t})\} The dynamics of Convotron can be expressed as follows:

We need to bound the RHS of the above equation. We have,

Combining the above equations and taking expectation over St−1S_{t-1}, we get

We set η=βmin⁡(1γ,ϵ∣∣w∗∣∣2B)\eta=\beta\min\left(\frac{1}{\gamma},\frac{\epsilon||w_{*}||^{2}}{B}\right) and break the analysis to two cases:

since at initialization ∣∣w1−w∗∣∣=∣∣w∗∣∣||w_{1}-w_{*}||=||w_{*}||. Setting T=O(1ηβlog⁡(1ϵδ))T=O\left(\frac{1}{\eta\beta}\log\left(\frac{1}{\epsilon\delta}\right)\right) and using Markov’s inequality, with probability 1−δ1-\delta, over the choice of STS_{T},

By using Lemma 2, we can get a bound on L(wT)≤ϵ∣∣w∗∣∣2L(w_{T})\leq\epsilon||w_{*}||^{2} by appropriately scaling ϵ\epsilon.

For the realizable (no noise) setting, that is, for all (x,y)∼D(x,y)\sim\mathcal{D}, y=fw∗(x)y=f_{w_{*}}(x), for some unknown w∗w_{*}, Convotron achieves faster convergence rates.

Observe that the dependence of ϵ\epsilon in the convergence rate is log⁡(1/ϵ)\log(1/\epsilon) for the realizable setting, compared to the 1/ϵ1/\epsilon dependence in the noisy setting.

Which Patch Structures are Easy to Learn?

In this section, we will show that the commonly used convolutional filters in practice (“patch and stride”) have good eigenvalues giving us fast convergence by Theorem 2. We will start with the 1D case and then subsequently extend the result for the 2D case.

Here we formally describe a patch and stride convolution in the one-dimensional setting. Consider a 1D image of dimension nn. Let the patch size be rr and stride be dd. Let the patches be indexed from 1 and let patch ii start at position (i−1)d+1(i-1)d+1 and be contiguous through position (i−1)d+r(i-1)d+r. The matrix PiP_{i} of dimension r×nr\times n corresponding to patch ii looks as follows,

where 0a×b0_{a\times b} indicates a matrix of dimension a×ba\times b with all zeros and IaI_{a} indicates the identity matrix of size aa.

Thus, the total number of patches is k=⌊n−rd⌋+1k=\lfloor\frac{n-r}{d}\rfloor+1. We will assume that n≥2r−1n\geq 2r-1 and r≥dr\geq d. The latter condition is to ensure there is some overlap, non-overlapping case, which is easier, is handled in the next section.

We will bound the extremal eigenvalues of P=∑i,j=1kPiPjTP=\sum_{i,j=1}^{k}P_{i}P_{j}^{T}. Simple algebra gives us the following structure for PP,

For understanding, we show the matrix structure for d=1d=1 and n≥2rn\geq 2r.

The following lemmas bound the extremal eigenvalues of PP.

Maximum eigenvalue of PP satisfies λmax⁡(P)≤k(p+1)−(p−p2)(p2+1)=O(kp)\lambda_{\max}(P)\leq k(p+1)-(p-p_{2})(p_{2}+1)=O(kp) where p=⌊r−1d⌋p=\lfloor\frac{r-1}{d}\rfloor and p2=⌊p2⌋p_{2}=\lfloor\frac{p}{2}\rfloor.

Using Theorem 1, we have λmax⁡(P)≤max⁡i(Pi,i+∑j≠i∣Pi,j∣)=max⁡i∑j=1kPi,j\lambda_{\max}(P)\leq\max_{i}\left(P_{i,i}+\sum_{j\neq i}|P_{i,j}|\right)=\max_{i}\sum_{j=1}^{k}P_{i,j}. Observe that PP is bisymmetric thus ∑j=1kPi,j=∑j=1kPr−i+1,j\sum_{j=1}^{k}P_{i,j}=\sum_{j=1}^{k}P_{r-i+1,j} and we can restrict to the top half of the matrix. The structure of PP indicates that in a fixed row, the diagonal entry is maximum and the non-zero entries decrease monotonically by 1 as we move away from the diagonal. Also, there can be at most p+1p+1 non-zero entries in any row. Thus the sum is maximized when there are p+1p+1 non-zero entries and the diagonal entry is the middle entry, that is at position p2d+1p_{2}d+1. By simple algebra,

Minimum eigenvalue of PP satisfies λmin⁡(P)≥0.5\lambda_{\min}(P)\geq 0.5.

We break the analysis into following two cases:

Case 1: d<r/2d<r/2 We can show that λmax⁡(P−1)≥2\lambda_{\max}(P^{-1})\geq 2 using the structure of PP (see Lemma 1 and 2). Since λmin⁡(P)=1/λmax⁡(P−1)\lambda_{\min}(P)=1/\lambda_{\max}(P^{-1}), we have λmin⁡(P)≥0.5\lambda_{\min}(P)\geq 0.5.

Case 2: d≥r/2d\geq r/2 In this case we directly bound the minimum eigenvalue of PP. Using Theorem 1, we know that λmin⁡(P)≥min⁡i(Pi,i−∑j≠i∣Pi,j∣)\lambda_{\min}(P)\geq\min_{i}\left(P_{i,i}-\sum_{j\neq i}|P_{i,j}|\right). For Pi,j≠0P_{i,j}\neq 0, ∣i−j∣=ad|i-j|=ad for some aa. The maximum value that ∣i−j∣|i-j| can take is r−1r-1 and since d≥r/2d\geq r/2, aa must be either 0 or 1. Also, for any ii, there exists a unique jj such that ∣i−j∣=d|i-j|=d since r/2≤d<rr/2\leq d<r, thus there are exactly 2 non-zero entries in each row of PP, Pi,iP_{i,i}. This gives us, for each ii, ∑j≠iPi,j=k−1\sum_{j\neq i}P_{i,j}=k-1. Thus, we get that λmin⁡(P)≥min⁡i(Pi,i−∣∑j≠iPi,j∣)=k−(k−1)=1\lambda_{\min}(P)\geq\min_{i}\left(P_{i,i}-\left|\sum_{j\neq i}P_{i,j}\right|\right)=k-(k-1)=1.

Combining both, we get the required result. ∎

1.2 Learning Result for 1D

Augmenting the above analysis with Theorem 2 gives us learnability of 1D convolution filters.

Combining the above Lemmas gives us that λmax⁡(P)=O(pk)=O(nr/d2)\lambda_{\max}(P)=O(pk)=O(nr/d^{2}) and λmin⁡(P)=Ω(1)\lambda_{\min}(P)=\Omega(1). Observe that λmin⁡(PΣ)≥λmin⁡(P)λmin⁡(Σ)\lambda_{\min}(P_{\Sigma})\geq\lambda_{\min}(P)\lambda_{\min}(\Sigma). Substituting these values in Theorem 2 gives us the desired result. ∎

Comparing with SGD, [BG17] showed that even for r=2r=2 and d=1d=1, Gradient descent can get stuck in a local minima with probability ≥1/4\geq 1/4.

2 2D Convolution

Here we formally define stride and patch convolutions in two dimensions. Consider a 2D image of dimension n1×n2n_{1}\times n_{2}. Let the patch size be r1×r2r_{1}\times r_{2} and stride in both directions be d1,d2d_{1},d_{2} respectively. Enumerate patches such that patch (i,j)(i,j) starts at position ((i−1)d1+1,(j−1)d2+1)((i-1)d_{1}+1,(j-1)d_{2}+1) and is a rectangle with diagonally opposite point ((i−1)d2+r1,(j−1)d2+r2)((i-1)d_{2}+r_{1},(j-1)d_{2}+r_{2}). Let k1=⌊n1−r1d1⌋+1k_{1}=\lfloor\frac{n_{1}-r_{1}}{d_{1}}\rfloor+1 and k2=⌊n2−r2d2⌋+1k_{2}=\lfloor\frac{n_{2}-r_{2}}{d_{2}}\rfloor+1. Let us vectorize the image row-wise into a n1n2n_{1}n_{2} dimension vector and enumerate each patch row-wise to get a r1r2r_{1}r_{2} dimensional vector.

Let Q(i,j)Q_{(i,j)} be the indicator matrix of dimension r1r2×n1n2r_{1}r_{2}\times n_{1}n_{2} with 1 at (a,b)(a,b) if the aath location of patch (i,j)(i,j) is bb. More formally, (Q(i,j))a,b=1(Q_{(i,j)})_{a,b}=1 for all a=pr2+q+1a=pr_{2}+q+1 for 0≤p<r10\leq p<r_{1}, 0≤q<r20\leq q<r_{2}, and b=((i−1)d1+p)n2+jd2+q+1b=((i-1)d_{1}+p)n_{2}+jd_{2}+q+1 else 0. Note that there are k1⋅k2k_{1}\cdot k_{2} patches in total with the corresponding patch matrices being Q(i,j)Q_{(i,j)} for 1≤i≤k1,1≤j≤k21\leq i\leq k_{1},1\leq j\leq k_{2}.

We will bound the extremal eigenvalues of Q=∑i,p=1k1∑j,q=1k2Q(i,j)Q(p,q)TQ=\sum_{i,p=1}^{k_{1}}\sum_{j,q=1}^{k_{2}}Q_{(i,j)}Q_{(p,q)}^{T}. Let Pi(1)P^{(1)}_{i}’s be the patch matrices corresponding to the 1D convolution for parameters n1,r1,d1n_{1},r_{1},d_{1} defined as in the previous section and let P(1)=∑i,j=1k1Pi(1)(Pj(1))TP^{(1)}=\sum_{i,j=1}^{k_{1}}P^{(1)}_{i}(P^{(1)}_{j})^{T}. Define Pi(2)P^{(2)}_{i}’s for 1≤i≤k21\leq i\leq k_{2} and P(2)P^{(2)} similarly with parameters n2,r2,d2n_{2},r_{2},d_{2} instead of n1,r1,d1n_{1},r_{1},d_{1}.

Q(i,j)=Pi(1)⊗Pj(2)Q_{(i,j)}=P^{(1)}_{i}\otimes P^{(2)}_{j}.

Intuitively Pi(1)P^{(1)}_{i} and Pj(2)P^{(2)}_{j} give the indices corresponding to the row and column of the 2D patch and the Kronecker product vectorizes it to give us the (i,j)(i,j)th patch. More formally, we will show that (Q(i,j))a,b=1(Q_{(i,j)})_{a,b}=1 iff (Pi(1)⊗Pj(2))a,b=1(P^{(1)}_{i}\otimes P^{(2)}_{j})_{a,b}=1.

Let a=pr2+q+1a=pr_{2}+q+1 with 0≤p<r10\leq p<r_{1}, 0≤q<r20\leq q<r_{2} and b=rn2+s+1b=rn_{2}+s+1 with 0≤r<n10\leq r<n_{1}, 0≤s<n20\leq s<n_{2}. Then, (Pi(1)⊗Pj(2))a,b=1(P^{(1)}_{i}\otimes P^{(2)}_{j})_{a,b}=1 iff (Pi(1))p,r=1(P^{(1)}_{i})_{p,r}=1 and (Pj(2))q,s=1(P^{(2)}_{j})_{q,s}=1. We know that (Pi(1))p,r=1(P^{(1)}_{i})_{p,r}=1 iff r=(i−1)d1+p+1r=(i-1)d_{1}+p+1 and (Pj(2))q,s=1(P^{(2)}_{j})_{q,s}=1 iff s=(j−1)d2+q+1s=(j-1)d_{2}+q+1. This gives us that b=((i−1)d1+p)n1+(j−1)d2+q+1b=((i-1)d_{1}+p)n_{1}+(j-1)d_{2}+q+1, which is the same condition for (Q(i,j))a,b=1(Q_{(i,j)})_{a,b}=1. Thus Q(i,j)=Pi(1)⊗Pj(2)Q_{(i,j)}=P^{(1)}_{i}\otimes P^{(2)}_{j}. ∎

We have λmin⁡(Q)≥0.25\lambda_{\min}(Q)\geq 0.25 and λmax⁡(Q)=O(k1p1k2p2)\lambda_{\max}(Q)=O(k_{1}p_{1}k_{2}p_{2}) where p1=⌊r1−1d1⌋p_{1}=\lfloor\frac{r_{1}-1}{d_{1}}\rfloor and p2=⌊r2−1d2⌋p_{2}=\lfloor\frac{r_{2}-1}{d_{2}}\rfloor.

Since Q=P(1)⊗P(2)Q=P^{(1)}\otimes P^{(2)} and Q,P(1),P(2)Q,P^{(1)},P^{(2)} are positive semi-definite, λmin⁡(Q)=λmin⁡(P)λmin⁡(P(2))\lambda_{\min}(Q)=\lambda_{\min}(P)\lambda_{\min}(P^{(2)}) and λmax⁡(Q)=λmax⁡(P(1))λmax⁡(P(2))\lambda_{\max}(Q)=\lambda_{\max}(P^{(1)})\lambda_{\max}(P^{(2)}). Using the lemmas from the previous section gives us the required result. ∎

Note that this technique can be extended to higher dimensional patch structures as well.

2.2 Learning Result for 2D

Similar to the 1D case, combining the above analysis with Theorem 2 gives us learnability of 2D convolution filters.

Lemma 5 gives us that λmax⁡(Q)=O(n1n2r1r2/(d1d2)2)\lambda_{\max}(Q)=O(n_{1}n_{2}r_{1}r_{2}/(d_{1}d_{2})^{2}) and λmin⁡(P)=Ω(1)\lambda_{\min}(P)=\Omega(1). Observe that λmin⁡(PΣ)≥λmin⁡(P)λmin⁡(Σ)\lambda_{\min}(P_{\Sigma})\geq\lambda_{\min}(P)\lambda_{\min}(\Sigma). Substituting these values in Theorem 2 gives us the desired result. ∎

Non-overlapping Patches are Easy

In this section, we will show that if there is one patch that does not overlap with any patch and the covariance matrix is identity then we can easily learn the filter even if the other patches have arbitrary overlaps. This includes the commonly used Gaussian assumption. WLOG we assume that P1P_{1} is the patch that does not overlap with any other patch implying P1PjT=PjTP1=0P_{1}P_{j}^{T}=P_{j}^{T}P_{1}=0 for all j≠1j\neq 1.

Observe that the algorithm ignores the directions of all other patches and yet succeeds. This indicates that with respect to a Gaussian distribution, in order to have an interesting patch structure (for one layer networks), it is necessary to avoid having even a single disjoint patch. The following theorem shows the convergence of Convotron-No-Overlap.

The proof follows the outline of the Convotron proof very closely. We use the same definitions as in the previous proof. We have,

The last equality follows since PiTP1=0P_{i}^{T}P_{1}=0 for all i≠1i\neq 1 and P1TP1P_{1}^{T}P_{1} is a permutation of identity.

Following the rest of the analysis for η\eta and TT as in the theorem statement gives us the required result. ∎

Experiments: SGD vs Convotron

To further support our theoretical findings, we empirically compare the performance of SGD (Algorithm 3) with our algorithm Convotron. We measure performance based on the failure probability, that is, the fraction of runs the algorithm fails to converge on randomly initialized runs (the randomness is over both the choice of initialization for SGD and the draws from the distribution). More formally, we say that the algorithm fails if the closeness in l2l_{2}-norm of the difference of the final weight vector obtained (wT)(w_{T}) and the true weight parameter (w∗w_{*}), that is, ∣∣wT−w∗∣∣||w_{T}-w_{*}|| is greater than a threshold θ\theta. We choose this measure because in practice, due to the high computation time of training neural networks, random restarts are expensive.

In the experiments, given a fixed true weight vector, for varying learning rates (increments of 0.010.01), we choose 50 random initializations and run the two algorithms with them as starting points. We plot the failure probability (θ=0.1\theta=0.1) with varying learning rate. Note that the lowest learning rate we use is 0.010.01 as making the learning rate too small requires high number of iterations for convergence for both algorithms.

We first test the performance on a simple 1D convolution case with (n,k,d,T)=(8,4,1,6000)(n,k,d,T)=(8,4,1,6000) and 2D case with (n1,n2,k1,k2,d1,d2,T)=(5,5,3,3,1,1,15000)(n_{1},n_{2},k_{1},k_{2},d_{1},d_{2},T)=(5,5,3,3,1,1,15000) on inputs drawn from a normalized (l2l_{2} norm 1) Gaussian distribution with identity covariance matrix. We adversarially choose a fixed weight vectorWe take the vector to be $inthe1Dcaseandnormalize.Thisweightvectorcanbeviewedasanedgedetectionfilter,thatis,countingthenumberoftimesimagegoesfromblack(negative)towhite(positive).(in the 1D case and normalize. This weight vector can be viewed as an edge detection filter, that is, counting the number of times image goes from black (negative) to white (positive). (l_{2}−norm1).Figure3(Top)showsthatSGDhasasmalldatadependentrangewhereitsucceedsbutmayfailwithalmost-norm 1). Figure 3 (Top) shows that SGD has a small data dependent range where it succeeds but may fail with almost0.5probabilityoutsidethisregionwhereasConvotronalwaysreturnsagoodsolutionforsmallenoughprobability outside this region whereas Convotron always returns a good solution for small enough\eta$ chosen according to Theorem 2. The failure points observed for SGD show the prevalence of bad local minima where SGD gets stuck.

For the second experiment, we choose a fixed weight vector for which SGD performs well with very high probability on a normalized Gaussian input distribution with identity covariance matrix (see Figure 3 (Bottom-left)). However, on choosing a different covariance matrix with higher condition number ∼60\sim 60, the performance of SGD worsens whereas Convotron always succeeds (see Figure 3 (Bottom-Right)). The covariance matrix is generated by choosing random matrices followed by symmetrizing them and adding cIcI for c>0c>0 to make the eigenvalues positive.

These experiments demonstrate that techniques for fine-tuning SGD’s learning rate are necessary, even for very simple architectures. In contrast, no fine-tuning is necessary for Convotron: the correct learning rate can be easily computed given the learner’s desired patch structure and estimate of the covariance martix.

Acknowledgments

We thank Jessica Hoffmann and Philipp Krähenbühl for useful discussions.

References

Appendix A Omitted Proofs

Since xx is drawn from a symmetric distribution we have Ex∼D[F(x)]=Ex∼D[F(−x)]E_{x\sim\mathcal{D}}[F(x)]=E_{x\sim\mathcal{D}}[F(-x)] for any function FF. Thus, we have

Observe that σ(c)−σ(−c)=(1−α)∣c∣+(1+α)c2−(1−α)∣a∣−(1+α)c2=(1+α)c\sigma(c)-\sigma(-c)=\frac{(1-\alpha)|c|+(1+\alpha)c}{2}-\frac{(1-\alpha)|a|-(1+\alpha)c}{2}=(1+\alpha)c. Substituting this in the above, we get the required result 2T1=(1+α)T22T_{1}=(1+\alpha)T_{2}.

A.2 Proof of Lemma 2

The first equality follows from using Lemma 1 and the last follows since for all ii, PiPiTP_{i}P_{i}^{T} is a permutation of the identity matrix by definition.

Using monotonicity of σ\sigma and Jensen’s inequality, we also have,

Combining the two above lemmas, we get the required result.

A.3 Proof of Lemma 3

The first inequality follows from using Jensen’s, the second inequality follows from the 1-Lipschitz property of σ\sigma, the third follows from observing that PiPiTP_{i}P_{i}^{T} is a PSD matrix and the last inequality follows since for all ii, λmax⁡(PiPiT)=1\lambda_{\max}(P_{i}P_{i}^{T})=1 since PiPiTP_{i}P_{i}^{T} is a permutation of the identity matrix.

Appendix B Properties of Patch Matrix P𝑃P

Let r=pd+qr=pd+q for some p≥0p\geq 0 and 1≤q≤d1\leq q\leq d.

For d<r/3d<r/3, P−1P^{-1} has the following form:

where α0=β+0.5\alpha_{0}=\beta+0.5, α1=ϕ+0.5\alpha_{1}=\phi+0.5, β=0.52k−p\beta=\frac{0.5}{2k-p} and ϕ=0.52k−p−1\phi=\frac{0.5}{2k-p-1}. Also, λmax⁡(P−1)≤2\lambda_{\max}(P^{-1})\leq 2.

We need to show that A=PP−1=IA=PP^{-1}=I. Observe that PP and P−1P^{-1} are bisymmetric, thus AA is centrosymmetric implying Ai,j=Ar−1−i,r−1−jA_{i,j}=A_{r-1-i,r-1-j}. Hence, we need to only prove that the lower triangular matrix matches II. We show the result for p>2p>2, as the same ideas apply for the other case.

To verify this, consider each diagonal entry,

d≤i≤⌈d/2⌉d\leq i\leq\lceil d/2\rceil: Ai,i=−0.5(k−1)+k−0.5(k−1)=1A_{i,i}=-0.5(k-1)+k-0.5(k-1)=1.

i∈{1,…,q}i\in\{1,\ldots,q\}: Ai,i=α0k−0.5(k−1)+β(k−p)=1A_{i,i}=\alpha_{0}k-0.5(k-1)+\beta\left(k-p\right)=1.

i∈{q+1,…,d}i\in\{q+1,\ldots,d\}: Ai,i=α1k−0.5(k−1)+ϕ(k−p−1)=1A_{i,i}=\alpha_{1}k-0.5(k-1)+\phi\left(k-p-1\right)=1.

For non-diagonal entries, that is, j≠ij\neq i,

d≤j≤⌈d/2⌉d\leq j\leq\lceil d/2\rceil: Ai,j=−0.5Pi,j−d+Pi,j−0.5Pi,j+dA_{i,j}=-0.5P_{i,j-d}+P_{i,j}-0.5P_{i,j+d}. If ∣i−j∣=ad|i-j|=ad then Ai,j=−0.5(k−a−1)+k−a−0.5(k−a+1)=0A_{i,j}=-0.5\left(k-a-1\right)+k-a-0.5\left(k-a+1\right)=0, else Pi,j=Pi,j−d=Pi,j+d=0  ⟹  Ai,j=0P_{i,j}=P_{i,j-d}=P_{i,j+d}=0\implies A_{i,j}=0.

j∈{1,…,q}j\in\{1,\ldots,q\}: Ai,j=α0Pi,j−0.5Pi,j+d+βPi,j+pdA_{i,j}=\alpha_{0}P_{i,j}-0.5P_{i,j+d}+\beta P_{i,j+pd}. Now if i−j=adi-j=ad, then Ai,j=α0(k−a)−0.5(k−a+1)+β(k−p+a)=0A_{i,j}=\alpha_{0}(k-a)-0.5(k-a+1)+\beta(k-p+a)=0 else Pi,j=Pi,j+d=Pi,j+pd=0  ⟹  Ai,j=0P_{i,j}=P_{i,j+d}=P_{i,j+pd}=0\implies A_{i,j}=0.

j∈{q+1,…,d}j\in\{q+1,\ldots,d\}: Ai,j=α1Pi,j−0.5Pi,j+d+βPi,j+pdA_{i,j}=\alpha_{1}P_{i,j}-0.5P_{i,j+d}+\beta P_{i,j+pd}. Now if i−j=adi-j=ad, then Ai,j=α1(k−a)−0.5(k−a+1)+ϕ(k−p+a+1)=0A_{i,j}=\alpha_{1}(k-a)-0.5(k-a+1)+\phi(k-p+a+1)=0 else Pi,j=Pi,j+d=Pi,j+pd=0  ⟹  Ai,j=0P_{i,j}=P_{i,j+d}=P_{i,j+pd}=0\implies A_{i,j}=0.

Using Theorem 1, we have λmax⁡(P−1)=max⁡i(Pi,i−1+∑j≠i∣Pi,j−1∣)\lambda_{\max}(P^{-1})=\max_{i}\left(P^{-1}_{i,i}+\sum_{j\neq i}|P^{-1}_{i,j}|\right). If q<dq<d, then λmax⁡(P−1)=max⁡(α0+0.5+β,α1+0.5+ϕ,1+0.5+0.5)=max⁡(2β+1,2ϕ+1,2)=2\lambda_{\max}(P^{-1})=\max(\alpha_{0}+0.5+\beta,\alpha_{1}+0.5+\phi,1+0.5+0.5)=\max(2\beta+1,2\phi+1,2)=2 as β,ϕ≤0.5\beta,\phi\leq 0.5 which follows from 2k−p−1≥12k-p-1\geq 1. Similarly, when q=dq=d, λmax⁡(P−1)=max⁡(α0+0.5+β,1+0.5+0.5)=max⁡(2β+1,2)=2\lambda_{\max}(P^{-1})=\max(\alpha_{0}+0.5+\beta,1+0.5+0.5)=\max(2\beta+1,2)=2. ∎

For r/3≤d<r/2r/3\leq d<r/2, P−1P^{-1} has the following form:

where α0=β+0.5\alpha_{0}=\beta+0.5, α1=k2k−1\alpha_{1}=\frac{k}{2k-1}, β=0.52k−2\beta=\frac{0.5}{2k-2} and ϕ=−k−12k−1\phi=-\frac{k-1}{2k-1}. Also, λmax⁡(P−1)≤2\lambda_{\max}(P^{-1})\leq 2.

Similar to the previous lemma, to verify this, consider each diagonal entry,

d≤i≤⌈d/2⌉d\leq i\leq\lceil d/2\rceil: Ai,i=−0.5(k−1)+k−0.5(k−1)=1A_{i,i}=-0.5(k-1)+k-0.5(k-1)=1.

i∈{1,…,q}i\in\{1,\ldots,q\}: Ai,i=α0k−0.5(k−1)+β(k−p)=1A_{i,i}=\alpha_{0}k-0.5(k-1)+\beta\left(k-p\right)=1.

i∈{q+1,…,d}i\in\{q+1,\ldots,d\}: Ai,i=α1k+ϕ(k−1)=1A_{i,i}=\alpha_{1}k+\phi(k-1)=1.

For non-diagonal entries, that is, j≠ij\neq i,

d≤j≤⌈d/2⌉d\leq j\leq\lceil d/2\rceil: Ai,j=−0.5Pi,j−d+Pi,j−0.5Pi,j+dA_{i,j}=-0.5P_{i,j-d}+P_{i,j}-0.5P_{i,j+d}. If ∣i−j∣=ad|i-j|=ad then Ai,j=−0.5(k−a−1)+k−a−0.5(k−a+1)=0A_{i,j}=-0.5\left(k-a-1\right)+k-a-0.5\left(k-a+1\right)=0, else Pi,j=Pi,j−d=Pi,j+d=0  ⟹  Ai,j=0P_{i,j}=P_{i,j-d}=P_{i,j+d}=0\implies A_{i,j}=0.

j∈{1,…,q}j\in\{1,\ldots,q\}: Ai,j=α0Pi,j−0.5Pi,j+d+βPi,j+2dA_{i,j}=\alpha_{0}P_{i,j}-0.5P_{i,j+d}+\beta P_{i,j+2d}. Now if i−j=adi-j=ad, then Ai,j=α0(k−a)−0.5(k−a+1)+β(k−2+a)=0A_{i,j}=\alpha_{0}(k-a)-0.5(k-a+1)+\beta(k-2+a)=0 else Pi,j=Pi,j+d=Pi,j+pd=0  ⟹  Ai,j=0P_{i,j}=P_{i,j+d}=P_{i,j+pd}=0\implies A_{i,j}=0.

j∈{q+1,…,d}j\in\{q+1,\ldots,d\}: Ai,j=α1Pi,j+ϕPi,j+dA_{i,j}=\alpha_{1}P_{i,j}+\phi P_{i,j+d}. Now if i−j=adi-j=ad, then a=1a=1, implying Ai,j=α1(k−1)+ϕk=0A_{i,j}=\alpha_{1}(k-1)+\phi k=0 else Pi,j=Pi,j+d=0  ⟹  Ai,j=0P_{i,j}=P_{i,j+d}=0\implies A_{i,j}=0.

Similar to the previous lemma, we have λmax⁡(P−1)=max⁡(α0+0.5+β,α1+∣ϕ∣,1+0.5+0.5)=max⁡(2β+1,1,2)=2\lambda_{\max}(P^{-1})=\max(\alpha_{0}+0.5+\beta,\alpha_{1}+|\phi|,1+0.5+0.5)=\max(2\beta+1,1,2)=2 as α1+∣ϕ∣=1\alpha_{1}+|\phi|=1 and β≤0.5\beta\leq 0.5 which follows from 2k−p−1≥12k-p-1\geq 1. ∎