Transfer learning-based method for automated ewaste recycling in smart cities

In plain words

Recycling old electronics works better when a machine can tell what a device is before it is taken apart. This paper looks at sorting smartphones from photos of their backs, where logos and camera layouts differ between brands.

Training an image classifier from nothing needs a lot of labelled images, which is hard to collect for a niche set of phones. The authors instead start from a network that was already trained on a large general photo collection, replace its final layers with ones that predict the phone's brand and model, and train only that part on their small set of photos.

They also make the small set go further by producing slightly changed copies of each photo (flipped, shifted, stretched, skewed), and they try a few training settings. The best combination reaches about 98% accuracy on their held-out photos, but that number comes from the authors' own small dataset.

TL;DR

• Transfer learning: an ImageNet-pretrained AlexNet is adapted to 12 smartphone classes from 6 brands §4, p. 5.
• Dataset: 650 RGB back-side photos collected manually from the internet, split 80/20 §4, p. 5.
• Best setting: SGD with momentum (SGDM), learning rate 3e-4, mini-batch 64, at most 30 epochs, with 9x augmentation §5.1 §5.2 §5.3.
• Reported accuracy: 98.3505 ± 2.0155 over seven trials, against 95.2577 ± 3.0427 (Adam) and 93.1959 ± 3.8635 (RMSProp) tab. 1.
• No official code; no held-out test set is described (see Pitfalls).

Problem

Electronic waste is sorted by hand or with fixed rules. The paper argues that automated, AI-based classification of devices would reduce sorting errors and labour, and it takes smartphones as the case study §1.1 §1.2 §1.3. It also notes that training a convolutional network from scratch needs a long training time and a large dataset, and that the authors wanted to avoid both §4, p. 4. The paper's stated assumption is that a small dataset of phone-back photos is enough once a pretrained network is reused §4, p. 4.

Method

The approach is supervised transfer learning on a convolutional network §3 §4, p. 4:

1. Start from AlexNet with weights pretrained on ImageNet §3, p. 4. The paper describes AlexNet's five convolutional layers and three fully connected layers and the ReLU, pooling and dropout components §3, p. 4 fig. 3.
2. Replace the last three layers with new ones that output 12 values, one per class §4, p. 5.
3. Freeze the pretrained weights of the earlier layers. The paper's wording on which layers stay frozen is inconsistent (see Pitfalls) §4, p. 5.
4. Split the photos 80% training and 20% validation §4, p. 5.
5. Augment the training images with eight transformed variants of each photo, giving 9 images per original photo §5.1, p. 5 fig. 4.
6. Train with SGD with momentum, learning rate 3e-4, mini-batch 64, and reshuffling every epoch, for at most 30 epochs §4, p. 5 §5.2, p. 6 §5.3, p. 7.
7. Report accuracy on the validation split, as the mean and standard deviation over seven trials per optimizer §5.2, p. 6 tab. 1.

The paper gives no equations and no algorithm box. The procedure is written out below in pseudocode, with every step taken from the text. The early-stop rule is only given as the epoch cap.

# Fine-tune a pretrained AlexNet on phone-back photos  [§4, p. 5]
# Input:  labelled photos, 12 classes
# Output: trained classifier, 12 outputs

def fine_tune(photos, labels):
    # Load AlexNet with ImageNet weights  [§3, p. 4]
    model = load_pretrained_alexnet()

    # Replace the last three layers with 12-class outputs  [§4, p. 5]
    model = replace_last_three_layers(model, num_classes=12)

    # Freeze pretrained weights; which layers stay frozen is ambiguous
    # (see Pitfalls)  [§4, p. 5]
    model = freeze_pretrained_layers(model)

    # Split 80% train, 20% validation  [§4, p. 5]
    train_set, val_set = split(photos, labels, train_fraction=0.8)

    # Nine versions of each training image: original plus eight
    # augmented copies  [§5.1, p. 5] [fig. 4]
    train_set = augment(train_set, copies_per_image=9)

    for epoch in range(30):  # maximum 30 epochs  [§4, p. 5]
        # Reshuffle the training set before every epoch  [§4, p. 5]
        for batch in minibatches(shuffle(train_set), batch_size=64):
            # SGD with momentum, learning rate 3e-4  [§5.2] [§5.3]
            loss = cross_entropy(model(batch.images), batch.labels)
            model = sgdm_update(model, loss, learning_rate=3e-4)

        # Accuracy on the validation split  [§5.2, p. 6]
        accuracy = evaluate(model, val_set)

    return model

Augmentation operations and ranges, as written §5.1, p. 5:

• Random X reflection, random Y reflection.
• Random X translation, random Y translation, with range [-50, 50] pixels.
• Random X scale, random Y scale, with scale range [0.9, 1.1].
• Random X shear, random Y shear, with range [-50, 50] pixels as written.

Hyperparameters

Results

Author-reported. Setup: 650-image smartphone dataset, 12 classes, AlexNet fine-tuned, validation split of 20%, MATLAB on one RTX 2070S §4, p. 5. The baselines are the other two optimizers, with the same learning rate of 3e-4 §5.2, p. 6. The paper reports no comparison against a model trained from scratch and no comparison without augmentation with a number §5.1, p. 5.

Paper vs code

No official code. Searched: the PDF's text, abstract and footnotes contain no code link. GitHub repository search for "ewaste transfer learning smartphone", "smartphone waste transfer learning", "ewaste recycling", and "e-waste smartphone classification alexnet" found no repository by the authors. The first author's GitHub account (NermeenAbouBaker) has eight repositories, none about this method; they were checked by name and description, not by reading the code.

Pitfalls

• Reported accuracy is validation accuracy. The paper splits off 20% as the validation set, checks performance on it, and selects the optimizer and learning rate on the same split §4, p. 5 §5.2, p. 6. No separate test set is described. The 98% figure is therefore optimistic; an implementer should hold out a test set.
• Augmentation may leak between splits. The paper does not say whether the 9x augmentation is applied before or after the 80/20 split §4, p. 5 §5.1, p. 5. If it is applied first, augmented copies of one photo can land in both training and validation sets, which inflates accuracy.
• Dataset size arithmetic. 650 original photos times nine gives 5850, which matches the paper's count §5.1, p. 5. Whether the 5850 are split 80/20, or the 650 are split and then augmented, is the leakage question above.
• Frozen layers are described inconsistently. The paper says the network is "initialized with frozen pre-trained weights for all layers except the last three" §4, p. 5, and then that "the learnable weights of AlexNet are frozen in the fully connected layer" §4, p. 5. These conflict. Decide explicitly which layers are trainable and document it.
• Shear range units. Shear is given in pixels, "[-50 50] pixel", the same as the translation range §5.1, p. 5. Shear is normally a unitless factor. The intended units are not stated.
• Augmentation benefit is not measured. Section 5.1 describes augmentation and shows examples fig. 4 but reports no accuracy with and without it. The claim that augmentation avoids overfitting is therefore unsupported by a number in the paper.
• Learning-rate choice is by eye. The sweep is judged from a box plot, and the authors say they "tried to guess" the rate by watching the loss §5.3, p. 7. The number of trials per learning rate is not stated; only the optimizer comparison states seven trials §5.2, p. 6.
• Differences in spread. The SGDM and Adam means differ by about 3 points, while the standard deviations are about 2 and 3 points tab. 1. No significance test is reported.
• Learning-rate schedule and momentum are missing. Neither the momentum coefficient nor a schedule is stated. Early stopping is described only as the epoch cap of 30 §4, p. 5.
• Dataset availability and image rights. The photos were collected from the internet §4, p. 5. The paper does not state a dataset license, a release, or permission for the images. Do not redistribute the photos; obtain the data independently if reproducing.
• Framework. Everything was done in MATLAB R2020b with CUDA on one consumer GPU §4, p. 5. Porting to another framework requires choosing the AlexNet weights and the input preprocessing; neither is stated.
• Date mismatch. The arXiv metadata gives a published date of 2026-06-22. The PDF is the EAI Endorsed Transactions version, with the dates "received 19 March 2021, accepted 12 April 2021, published 16 April 2021" p. 1. The card follows the arXiv metadata for published; the PDF is the source for every citation.

Lineage

• Builds on: AlexNet [ref. 30] and the transfer learning approach it cites [ref. 29] for reusing pretrained features §4, p. 4.
• The fine-tuning procedure follows the same authors' earlier work on smartphone recycling, cited as "previous work [8]" §4, p. 5. Reference 8 is "A feature-fusion transfer learning method as a basis to support automated smartphone recycling in a circular smart city" (2020) [ref. 8]. It has no card here.
• Compared against: Adam and RMSProp optimizers §5.2, p. 6. No other published method is compared.
• Superseded by: not known from content on this wiki.