Compute Distance with Matrix

cs231n assignment1

Posted by cyc1am3n on July 06, 2018

이 포스트는 knn (k-nearest neighbors) 구현에 필요한 test point와 training point의 L2 distance matrix를 python을 통해 구하는 방법을 다룰 것이다.

What is L2 Distance?

L2 distance(Euclidean distance)는 유클리드 좌표계에서 두 점 사이의 직선 거리를 의미한다. 유클리드 공간에서 두 점 p = (p1, p2, …, pn)와 q = (q1, q2, …, qn)의 L2 distance는 다음과 같다.

그림1

With Two loops

제일 간단한 방법으로, 모든 test data와 training data를 하나하나 비교하는 방법이라 vectorization 을 사용하지 않으며 시간이 오래 걸린다.

def compute_distances_two_loops(self, X):
    num_test = X.shape[0]
    num_train = self.X_train.shape[0]
    dists = np.zeros((num_test, num_train))
    for i in range(num_test):
         for j in range(num_train):
            sub = np.subtract(X[i], self.X_train[j])
            square = np.square(sub)
            dists[i, j] = np.sqrt(np.sum(square))
    return dists

With One Loop

numpy.sum 함수를 이용해 vectorization을 했다. sum 함수에서 axis=1 일 때는 column을 합치게 된다.

def compute_distances_one_loop(self, X):
    num_test = X.shape[0]
    num_train = self.X_train.shape[0]
    dists = np.zeros((num_test, num_train))
    for i in range(num_test):
         dists[i, :] = np.sqrt(np.sum(np.square(self.X_train - X[i, :]), axis=1))
    return dists

Without loop

loop 없이 L2 distance를 구할 때 약간의 트릭이 필요한데,

(x - y)2 = x2 + y2 - 2xy

를 이용하는 것이다.

여기에서는 numpy.tile 함수를 사용했는데, 이는 square 후에 column으로 합친 행렬을 num_test 또는 num_train만큼 복사하는 함수이다.

tile을 통해 X2와 Y2을 더하고 X * Y를 X와 Y(train)을 transposing 한 Y.T를 dot product한 값을 더하면 L2 distance가 만들어진다.

실행 시간을 서로 비교해보면, 역시 loop가 줄어들수록 빨리 실행되는 것을 볼 수 있다.

def compute_distances_no_loops(self, X):
    num_test = X.shape[0]
    num_train = self.X_train.shape[0]
    dists = np.zeros((num_test, num_train))
    dists = np.sqrt(np.tile(np.sum(np.square(self.X_train), axis=1), (num_test, 1)) + np.tile(np.sum(np.square(X), axis=1), (num_train, 1)).T - 2 * np.dot(X, self.X_train.T))
    return dists